diff libmpcodecs/ad_spdif.c @ 34103:febe300dbfc0

S/PDIF passthrough decoder patch by Naoya OYAMA, naoya.oyama gmail com

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/libmpcodecs/ad_spdif.c	Wed Oct 12 17:23:08 2011 +0000
@@ -0,0 +1,311 @@
+/*
+ * 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 <string.h>
+
+#include "config.h"
+#include "mp_msg.h"
+#include "ad_internal.h"
+#include "libavformat/avformat.h"
+#include "libavcodec/avcodec.h"
+#include "libavutil/opt.h"
+
+static const ad_info_t info = {
+    "libavformat/spdifenc audio pass-through decoder.",
+    "spdif",
+    "Naoya OYAMA",
+    "Naoya OYAMA",
+    "For ALL hardware decoders"
+};
+
+LIBAD_EXTERN(spdif)
+
+#define FILENAME_SPDIFENC "spdif"
+#define OUTBUF_SIZE 65536
+struct spdifContext {
+    AVFormatContext *lavf_ctx;
+    int              iec61937_packet_size;
+    int              out_buffer_len;
+    int              out_buffer_size;
+    uint8_t         *out_buffer;
+    uint8_t          pb_buffer[OUTBUF_SIZE];
+};
+
+static int read_packet(void *p, uint8_t *buf, int buf_size)
+{
+    // spdifenc does not use read callback.
+    return 0;
+}
+
+static int write_packet(void *p, uint8_t *buf, int buf_size)
+{
+    int len;
+    struct spdifContext *ctx = p;
+
+    len = FFMIN(buf_size, ctx->out_buffer_size -ctx->out_buffer_len);
+    memcpy(&ctx->out_buffer[ctx->out_buffer_len], buf, len);
+    ctx->out_buffer_len += len;
+    return len;
+}
+
+static int64_t seek(void *p, int64_t offset, int whence)
+{
+    // spdifenc does not use seek callback.
+    return 0;
+}
+
+static int preinit(sh_audio_t *sh)
+{
+    sh->samplesize = 2;
+    return 1;
+}
+
+static int init(sh_audio_t *sh)
+{
+    int i, x, in_size, srate, bps, *dtshd_rate;
+    unsigned char *start;
+    double pts;
+    static const struct {
+        const char *name; enum CodecID id;
+    } fmt_id_type[] = {
+        { "aac" , CODEC_ID_AAC    },
+        { "ac3" , CODEC_ID_AC3    },
+        { "dca" , CODEC_ID_DTS    },
+        { "eac3", CODEC_ID_EAC3   },
+        { "mpa" , CODEC_ID_MP3    },
+        { "thd" , CODEC_ID_TRUEHD },
+        { NULL  , 0 }
+    };
+    AVFormatContext     *lavf_ctx  = NULL;
+    AVStream            *stream    = NULL;
+    const AVOption      *opt       = NULL;
+    struct spdifContext *spdif_ctx = NULL;
+
+    spdif_ctx = av_mallocz(sizeof(*spdif_ctx));
+    if (!spdif_ctx)
+        goto fail;
+    spdif_ctx->lavf_ctx = avformat_alloc_context();
+    if (!spdif_ctx->lavf_ctx)
+        goto fail;
+
+    sh->context = spdif_ctx;
+    lavf_ctx    = spdif_ctx->lavf_ctx;
+
+    lavf_ctx->oformat = av_guess_format(FILENAME_SPDIFENC, NULL, NULL);
+    if (!lavf_ctx->oformat)
+        goto fail;
+    lavf_ctx->priv_data = av_mallocz(lavf_ctx->oformat->priv_data_size);
+    if (!lavf_ctx->priv_data)
+        goto fail;
+    lavf_ctx->pb = avio_alloc_context(spdif_ctx->pb_buffer, OUTBUF_SIZE, 1, spdif_ctx,
+                            read_packet, write_packet, seek);
+    if (!lavf_ctx->pb)
+        goto fail;
+    stream = av_new_stream(lavf_ctx, 0);
+    if (!stream)
+        goto fail;
+    lavf_ctx->duration   = AV_NOPTS_VALUE;
+    lavf_ctx->start_time = AV_NOPTS_VALUE;
+    for (i = 0; fmt_id_type[i].name; i++) {
+        if (!strcmp(sh->codec->dll, fmt_id_type[i].name)) {
+            lavf_ctx->streams[0]->codec->codec_id = fmt_id_type[i].id;
+            break;
+        }
+    }
+    lavf_ctx->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
+    if (AVERROR_PATCHWELCOME == lavf_ctx->oformat->write_header(lavf_ctx)) {
+        mp_msg(MSGT_DECAUDIO,MSGL_INFO,
+               "This codec is not supported by spdifenc.\n");
+        goto fail;
+    }
+
+    // get sample_rate & bitrate from parser
+    bps = srate = 0;
+    x = ds_get_packet_pts(sh->ds, &start, &pts);
+    in_size = x;
+    if (x <= 0) {
+        pts = MP_NOPTS_VALUE;
+        x = 0;
+    }
+    ds_parse(sh->ds, &start, &x, pts, 0);
+    if (x == 0) { // not enough buffer
+        srate = 48000;    //fake value
+        bps   = 768000/8; //fake value
+    } else  if (sh->avctx) {
+        if (sh->avctx->sample_rate < 44100) {
+            mp_msg(MSGT_DECAUDIO,MSGL_INFO,
+                   "This stream sample_rate[%d Hz] may be broken. "
+                   "Force reset 48000Hz.\n",
+                   sh->avctx->sample_rate);
+            srate = 48000; //fake value
+        } else
+            srate = sh->avctx->sample_rate;
+        bps = sh->avctx->bit_rate/8;
+    } else {
+        ;
+    }
+    sh->ds->buffer_pos -= in_size;
+
+    switch (lavf_ctx->streams[0]->codec->codec_id) {
+    case CODEC_ID_AAC:
+        spdif_ctx->iec61937_packet_size = 16384;
+        sh->sample_format               = AF_FORMAT_IEC61937_LE;
+        sh->samplerate                  = srate;
+        sh->channels                    = 2;
+        sh->i_bps                       = bps;
+        break;
+    case CODEC_ID_AC3:
+        spdif_ctx->iec61937_packet_size = 6144;
+        sh->sample_format               = AF_FORMAT_IEC61937_LE;
+        sh->samplerate                  = srate;
+        sh->channels                    = 2;
+        sh->i_bps                       = bps;
+        break;
+    case CODEC_ID_DTS: // FORCE USE DTS-HD
+        opt = av_opt_find(&lavf_ctx->oformat->priv_class,
+                          "dtshd_rate", NULL, 0, 0);
+        if (!opt)
+            goto fail;
+        dtshd_rate                      = (int*)(((uint8_t*)lavf_ctx->priv_data) +
+                                          opt->offset);
+        *dtshd_rate                     = 192000*4;
+        spdif_ctx->iec61937_packet_size = 32768;
+        sh->sample_format               = AF_FORMAT_IEC61937_LE;
+        sh->samplerate                  = 192000; // DTS core require 48000
+        sh->channels                    = 2*4;
+        sh->i_bps                       = bps;
+        break;
+    case CODEC_ID_EAC3:
+        spdif_ctx->iec61937_packet_size = 24576;
+        sh->sample_format               = AF_FORMAT_IEC61937_LE;
+        sh->samplerate                  = 192000;
+        sh->channels                    = 2;
+        sh->i_bps                       = bps;
+        break;
+    case CODEC_ID_MP3:
+        spdif_ctx->iec61937_packet_size = 4608;
+        sh->sample_format               = AF_FORMAT_MPEG2;
+        sh->samplerate                  = srate;
+        sh->channels                    = 2;
+        sh->i_bps                       = bps;
+        break;
+    case CODEC_ID_TRUEHD:
+        spdif_ctx->iec61937_packet_size = 61440;
+        sh->sample_format               = AF_FORMAT_IEC61937_LE;
+        sh->samplerate                  = 192000;
+        sh->channels                    = 8;
+        sh->i_bps                       = bps;
+        break;
+    default:
+        break;
+    }
+
+    return 1;
+
+fail:
+    uninit(sh);
+    return 0;
+}
+
+static int decode_audio(sh_audio_t *sh, unsigned char *buf,
+                        int minlen, int maxlen)
+{
+    struct spdifContext *spdif_ctx = sh->context;
+    AVFormatContext     *lavf_ctx  = spdif_ctx->lavf_ctx;
+    AVPacket            pkt;
+    double              pts;
+    int                 ret, in_size, consumed, x;
+    unsigned char       *start = NULL;
+
+    consumed = spdif_ctx->out_buffer_len  = 0;
+    spdif_ctx->out_buffer_size = maxlen;
+    spdif_ctx->out_buffer      = buf;
+    while (spdif_ctx->out_buffer_len + spdif_ctx->iec61937_packet_size < maxlen
+           && spdif_ctx->out_buffer_len < minlen) {
+        if (sh->ds->eof)
+            break;
+        x = ds_get_packet_pts(sh->ds, &start, &pts);
+        if (x <= 0) {
+            x = 0;
+            ds_parse(sh->ds, &start, &x, MP_NOPTS_VALUE, 0);
+            if (x == 0)
+                continue; // END_NOT_FOUND
+            in_size = x;
+        } else {
+            in_size = x;
+            consumed = ds_parse(sh->ds, &start, &x, pts, 0);
+            if (x == 0) {
+                mp_msg(MSGT_DECAUDIO,MSGL_V,
+                       "start[%p] pkt.size[%d] in_size[%d] consumed[%d] x[%d].\n",
+                       start, pkt.size, in_size, consumed, x);
+                continue; // END_NOT_FOUND
+            }
+            sh->ds->buffer_pos -= in_size - consumed;
+        }
+        av_init_packet(&pkt);
+        pkt.data = start;
+        pkt.size = x;
+        mp_msg(MSGT_DECAUDIO,MSGL_V,
+               "start[%p] pkt.size[%d] in_size[%d] consumed[%d] x[%d].\n",
+               start, pkt.size, in_size, consumed, x);
+        if (pts != MP_NOPTS_VALUE) {
+            sh->pts       = pts;
+            sh->pts_bytes = 0;
+        }
+        ret = lavf_ctx->oformat->write_packet(lavf_ctx, &pkt);
+        if (ret < 0)
+            break;
+    }
+    sh->pts_bytes += spdif_ctx->out_buffer_len;
+    return spdif_ctx->out_buffer_len;
+}
+
+static int control(sh_audio_t *sh, int cmd, void* arg, ...)
+{
+    unsigned char *start;
+    double pts;
+
+    switch (cmd) {
+    case ADCTRL_RESYNC_STREAM:
+    case ADCTRL_SKIP_FRAME:
+        ds_get_packet_pts(sh->ds, &start, &pts);
+        return CONTROL_TRUE;
+    }
+    return CONTROL_UNKNOWN;
+}
+
+static void uninit(sh_audio_t *sh)
+{
+    struct spdifContext *spdif_ctx = sh->context;
+    AVFormatContext     *lavf_ctx  = spdif_ctx->lavf_ctx;
+
+    if (lavf_ctx) {
+        if (lavf_ctx->oformat)
+            lavf_ctx->oformat->write_trailer(lavf_ctx);
+        av_freep(&lavf_ctx->pb);
+        if (lavf_ctx->streams) {
+            av_freep(&lavf_ctx->streams[0]->codec);
+            av_freep(&lavf_ctx->streams[0]->info);
+            av_freep(&lavf_ctx->streams[0]);
+        }
+        av_freep(&lavf_ctx->streams);
+        av_freep(&lavf_ctx->priv_data);
+    }
+    av_freep(&lavf_ctx);
+    av_freep(&spdif_ctx);
+}