libmpcodecs/ad_ffmpeg.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <unistd.h>
   4
   5 #include "config.h"
   6 #include "mp_msg.h"
   7 #include "help_mp.h"
   8
   9 #include "ad_internal.h"
  10 #include "libaf/reorder_ch.h"
  11
  12 #include "mpbswap.h"
  13
  14 static ad_info_t info =
  15 {
  16         "FFmpeg/libavcodec audio decoders",
  17         "ffmpeg",
  18         "Nick Kurshev",
  19         "ffmpeg.sf.net",
  20         ""
  21 };
  22
  23 LIBAD_EXTERN(ffmpeg)
  24
  25 #define assert(x)
  26
  27 #include "libavcodec/avcodec.h"
  28
  29 extern int avcodec_initialized;
  30
  31 static int preinit(sh_audio_t *sh)
  32 {
  33   sh->audio_out_minsize=AVCODEC_MAX_AUDIO_FRAME_SIZE;
  34   return 1;
  35 }
  36
  37 static int init(sh_audio_t *sh_audio)
  38 {
  39     int x;
  40     AVCodecContext *lavc_context;
  41     AVCodec *lavc_codec;
  42
  43     mp_msg(MSGT_DECAUDIO,MSGL_V,"FFmpeg's libavcodec audio codec\n");
  44     if(!avcodec_initialized){
  45       avcodec_init();
  46       avcodec_register_all();
  47       avcodec_initialized=1;
  48     }
  49
  50     lavc_codec = (AVCodec *)avcodec_find_decoder_by_name(sh_audio->codec->dll);
  51     if(!lavc_codec){
  52         mp_msg(MSGT_DECAUDIO,MSGL_ERR,MSGTR_MissingLAVCcodec,sh_audio->codec->dll);
  53         return 0;
  54     }
  55
  56     lavc_context = avcodec_alloc_context();
  57     sh_audio->context=lavc_context;
  58
  59     lavc_context->sample_rate = sh_audio->samplerate;
  60     lavc_context->bit_rate = sh_audio->i_bps * 8;
  61     if(sh_audio->wf){
  62         lavc_context->channels = sh_audio->wf->nChannels;
  63         lavc_context->sample_rate = sh_audio->wf->nSamplesPerSec;
  64         lavc_context->bit_rate = sh_audio->wf->nAvgBytesPerSec * 8;
  65         lavc_context->block_align = sh_audio->wf->nBlockAlign;
  66         lavc_context->bits_per_coded_sample = sh_audio->wf->wBitsPerSample;
  67     }
  68     lavc_context->request_channels = audio_output_channels;
  69     lavc_context->codec_tag = sh_audio->format; //FOURCC
  70     lavc_context->codec_id = lavc_codec->id; // not sure if required, imho not --A'rpi
  71
  72     /* alloc extra data */
  73     if (sh_audio->wf && sh_audio->wf->cbSize > 0) {
  74         lavc_context->extradata = av_mallocz(sh_audio->wf->cbSize + FF_INPUT_BUFFER_PADDING_SIZE);
  75         lavc_context->extradata_size = sh_audio->wf->cbSize;
  76         memcpy(lavc_context->extradata, (char *)sh_audio->wf + sizeof(WAVEFORMATEX),
  77                lavc_context->extradata_size);
  78     }
  79
  80     // for QDM2
  81     if (sh_audio->codecdata_len && sh_audio->codecdata && !lavc_context->extradata)
  82     {
  83         lavc_context->extradata = av_malloc(sh_audio->codecdata_len);
  84         lavc_context->extradata_size = sh_audio->codecdata_len;
  85         memcpy(lavc_context->extradata, (char *)sh_audio->codecdata,
  86                lavc_context->extradata_size);
  87     }
  88
  89     /* open it */
  90     if (avcodec_open(lavc_context, lavc_codec) < 0) {
  91         mp_msg(MSGT_DECAUDIO,MSGL_ERR, MSGTR_CantOpenCodec);
  92         return 0;
  93     }
  94    mp_msg(MSGT_DECAUDIO,MSGL_V,"INFO: libavcodec init OK!\n");
  95
  96 //   printf("\nFOURCC: 0x%X\n",sh_audio->format);
  97    if(sh_audio->format==0x3343414D){
  98        // MACE 3:1
  99        sh_audio->ds->ss_div = 2*3; // 1 samples/packet
 100        sh_audio->ds->ss_mul = 2*sh_audio->wf->nChannels; // 1 byte*ch/packet
 101    } else
 102    if(sh_audio->format==0x3643414D){
 103        // MACE 6:1
 104        sh_audio->ds->ss_div = 2*6; // 1 samples/packet
 105        sh_audio->ds->ss_mul = 2*sh_audio->wf->nChannels; // 1 byte*ch/packet
 106    }
 107
 108    // Decode at least 1 byte:  (to get header filled)
 109    x=decode_audio(sh_audio,sh_audio->a_buffer,1,sh_audio->a_buffer_size);
 110    if(x>0) sh_audio->a_buffer_len=x;
 111
 112   sh_audio->channels=lavc_context->channels;
 113   sh_audio->samplerate=lavc_context->sample_rate;
 114   sh_audio->i_bps=lavc_context->bit_rate/8;
 115   switch (lavc_context->sample_fmt) {
 116       case SAMPLE_FMT_U8:  sh_audio->sample_format = AF_FORMAT_U8;       break;
 117       case SAMPLE_FMT_S16: sh_audio->sample_format = AF_FORMAT_S16_NE;   break;
 118       case SAMPLE_FMT_S32: sh_audio->sample_format = AF_FORMAT_S32_NE;   break;
 119       case SAMPLE_FMT_FLT: sh_audio->sample_format = AF_FORMAT_FLOAT_NE; break;
 120       default:
 121           mp_msg(MSGT_DECAUDIO, MSGL_FATAL, "Unsupported sample format\n");
 122           return 0;
 123   }
 124   if(sh_audio->wf){
 125       // If the decoder uses the wrong number of channels all is lost anyway.
 126       // sh_audio->channels=sh_audio->wf->nChannels;
 127       if (sh_audio->wf->nSamplesPerSec)
 128       sh_audio->samplerate=sh_audio->wf->nSamplesPerSec;
 129       if (sh_audio->wf->nAvgBytesPerSec)
 130       sh_audio->i_bps=sh_audio->wf->nAvgBytesPerSec;
 131   }
 132   sh_audio->samplesize=af_fmt2bits(sh_audio->sample_format)/ 8;
 133   return 1;
 134 }
 135
 136 static void uninit(sh_audio_t *sh)
 137 {
 138     AVCodecContext *lavc_context = sh->context;
 139
 140     if (avcodec_close(lavc_context) < 0)
 141         mp_msg(MSGT_DECVIDEO, MSGL_ERR, MSGTR_CantCloseCodec);
 142     av_freep(&lavc_context->extradata);
 143     av_freep(&lavc_context);
 144 }
 145
 146 static int control(sh_audio_t *sh,int cmd,void* arg, ...)
 147 {
 148     AVCodecContext *lavc_context = sh->context;
 149     switch(cmd){
 150     case ADCTRL_RESYNC_STREAM:
 151         avcodec_flush_buffers(lavc_context);
 152     return CONTROL_TRUE;
 153     }
 154     return CONTROL_UNKNOWN;
 155 }
 156
 157 static int decode_audio(sh_audio_t *sh_audio,unsigned char *buf,int minlen,int maxlen)
 158 {
 159     unsigned char *start=NULL;
 160     int y,len=-1;
 161     while(len<minlen){
 162         int len2=maxlen;
 163         double pts;
 164         int x=ds_get_packet_pts(sh_audio->ds,&start, &pts);
 165         if(x<=0) break; // error
 166         if (pts != MP_NOPTS_VALUE) {
 167             sh_audio->pts = pts;
 168             sh_audio->pts_bytes = 0;
 169         }
 170         y=avcodec_decode_audio2(sh_audio->context,(int16_t*)buf,&len2,start,x);
 171 //printf("return:%d samples_out:%d bitstream_in:%d sample_sum:%d\n", y, len2, x, len); fflush(stdout);
 172         if(y<0){ mp_msg(MSGT_DECAUDIO,MSGL_V,"lavc_audio: error\n");break; }
 173         if(y<x) sh_audio->ds->buffer_pos+=y-x;  // put back data (HACK!)
 174         if(len2>0){
 175           if (((AVCodecContext *)sh_audio->context)->channels >= 5) {
 176             int src_ch_layout = AF_CHANNEL_LAYOUT_MPLAYER_DEFAULT;
 177             const char *codec=((AVCodecContext*)sh_audio->context)->codec->name;
 178             if (!strcasecmp(codec, "ac3")
 179                 || !strcasecmp(codec, "eac3"))
 180               src_ch_layout = AF_CHANNEL_LAYOUT_LAVC_AC3_DEFAULT;
 181             else if (!strcasecmp(codec, "dca"))
 182               src_ch_layout = AF_CHANNEL_LAYOUT_LAVC_DCA_DEFAULT;
 183             else if (!strcasecmp(codec, "libfaad")
 184                 || !strcasecmp(codec, "mpeg4aac"))
 185               src_ch_layout = AF_CHANNEL_LAYOUT_AAC_DEFAULT;
 186             else if (!strcasecmp(codec, "liba52"))
 187               src_ch_layout = AF_CHANNEL_LAYOUT_LAVC_LIBA52_DEFAULT;
 188             else if (!strcasecmp(codec, "vorbis"))
 189               src_ch_layout = AF_CHANNEL_LAYOUT_VORBIS_DEFAULT;
 190             else if (!strcasecmp(codec, "flac"))
 191               src_ch_layout = AF_CHANNEL_LAYOUT_FLAC_DEFAULT;
 192             else
 193               src_ch_layout = AF_CHANNEL_LAYOUT_MPLAYER_DEFAULT;
 194             reorder_channel_nch(buf, src_ch_layout,
 195                                 AF_CHANNEL_LAYOUT_MPLAYER_DEFAULT,
 196                                 ((AVCodecContext *)sh_audio->context)->channels,
 197                                 len2 / 2, 2);
 198           }
 199           //len=len2;break;
 200           if(len<0) len=len2; else len+=len2;
 201           buf+=len2;
 202           maxlen -= len2;
 203           sh_audio->pts_bytes += len2;
 204         }
 205         mp_dbg(MSGT_DECAUDIO,MSGL_DBG2,"Decoded %d -> %d  \n",y,len2);
 206     }
 207   return len;
 208 }