git.sesse.net Git - ffmpeg/blob - libavcodec/ra144dec.c

   1 /*
   2  * Real Audio 1.0 (14.4K)
   3  *
   4  * Copyright (c) 2008 Vitor Sessak
   5  * Copyright (c) 2003 Nick Kurshev
   6  *     Based on public domain decoder at http://www.honeypot.net/audio
   7  *
   8  * This file is part of FFmpeg.
   9  *
  10  * FFmpeg is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU Lesser General Public
  12  * License as published by the Free Software Foundation; either
  13  * version 2.1 of the License, or (at your option) any later version.
  14  *
  15  * FFmpeg is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  * Lesser General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Lesser General Public
  21  * License along with FFmpeg; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  23  */
  24
  25 #include "libavutil/channel_layout.h"
  26 #include "libavutil/intmath.h"
  27 #include "avcodec.h"
  28 #include "get_bits.h"
  29 #include "internal.h"
  30 #include "ra144.h"
  31
  32
  33 static av_cold int ra144_decode_init(AVCodecContext * avctx)
  34 {
  35     RA144Context *ractx = avctx->priv_data;
  36
  37     ractx->avctx = avctx;
  38
  39     ractx->lpc_coef[0] = ractx->lpc_tables[0];
  40     ractx->lpc_coef[1] = ractx->lpc_tables[1];
  41
  42     avctx->channels       = 1;
  43     avctx->channel_layout = AV_CH_LAYOUT_MONO;
  44     avctx->sample_fmt     = AV_SAMPLE_FMT_S16;
  45
  46     avcodec_get_frame_defaults(&ractx->frame);
  47     avctx->coded_frame = &ractx->frame;
  48
  49     return 0;
  50 }
  51
  52 static void do_output_subblock(RA144Context *ractx, const uint16_t  *lpc_coefs,
  53                                int gval, GetBitContext *gb)
  54 {
  55     int cba_idx = get_bits(gb, 7); // index of the adaptive CB, 0 if none
  56     int gain    = get_bits(gb, 8);
  57     int cb1_idx = get_bits(gb, 7);
  58     int cb2_idx = get_bits(gb, 7);
  59
  60     ff_subblock_synthesis(ractx, lpc_coefs, cba_idx, cb1_idx, cb2_idx, gval,
  61                           gain);
  62 }
  63
  64 /** Uncompress one block (20 bytes -> 160*2 bytes). */
  65 static int ra144_decode_frame(AVCodecContext * avctx, void *data,
  66                               int *got_frame_ptr, AVPacket *avpkt)
  67 {
  68     const uint8_t *buf = avpkt->data;
  69     int buf_size = avpkt->size;
  70     static const uint8_t sizes[LPC_ORDER] = {6, 5, 5, 4, 4, 3, 3, 3, 3, 2};
  71     unsigned int refl_rms[NBLOCKS];           // RMS of the reflection coefficients
  72     uint16_t block_coefs[NBLOCKS][LPC_ORDER]; // LPC coefficients of each sub-block
  73     unsigned int lpc_refl[LPC_ORDER];         // LPC reflection coefficients of the frame
  74     int i, j;
  75     int ret;
  76     int16_t *samples;
  77     unsigned int energy;
  78
  79     RA144Context *ractx = avctx->priv_data;
  80     GetBitContext gb;
  81
  82     /* get output buffer */
  83     ractx->frame.nb_samples = NBLOCKS * BLOCKSIZE;
  84     if ((ret = ff_get_buffer(avctx, &ractx->frame)) < 0) {
  85         av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
  86         return ret;
  87     }
  88     samples = (int16_t *)ractx->frame.data[0];
  89
  90     if(buf_size < FRAMESIZE) {
  91         av_log(avctx, AV_LOG_ERROR,
  92                "Frame too small (%d bytes). Truncated file?\n", buf_size);
  93         *got_frame_ptr = 0;
  94         return buf_size;
  95     }
  96     init_get_bits(&gb, buf, FRAMESIZE * 8);
  97
  98     for (i = 0; i < LPC_ORDER; i++)
  99         lpc_refl[i] = ff_lpc_refl_cb[i][get_bits(&gb, sizes[i])];
 100
 101     ff_eval_coefs(ractx->lpc_coef[0], lpc_refl);
 102     ractx->lpc_refl_rms[0] = ff_rms(lpc_refl);
 103
 104     energy = ff_energy_tab[get_bits(&gb, 5)];
 105
 106     refl_rms[0] = ff_interp(ractx, block_coefs[0], 1, 1, ractx->old_energy);
 107     refl_rms[1] = ff_interp(ractx, block_coefs[1], 2,
 108                             energy <= ractx->old_energy,
 109                             ff_t_sqrt(energy*ractx->old_energy) >> 12);
 110     refl_rms[2] = ff_interp(ractx, block_coefs[2], 3, 0, energy);
 111     refl_rms[3] = ff_rescale_rms(ractx->lpc_refl_rms[0], energy);
 112
 113     ff_int_to_int16(block_coefs[3], ractx->lpc_coef[0]);
 114
 115     for (i=0; i < NBLOCKS; i++) {
 116         do_output_subblock(ractx, block_coefs[i], refl_rms[i], &gb);
 117
 118         for (j=0; j < BLOCKSIZE; j++)
 119             *samples++ = av_clip_int16(ractx->curr_sblock[j + 10] << 2);
 120     }
 121
 122     ractx->old_energy = energy;
 123     ractx->lpc_refl_rms[1] = ractx->lpc_refl_rms[0];
 124
 125     FFSWAP(unsigned int *, ractx->lpc_coef[0], ractx->lpc_coef[1]);
 126
 127     *got_frame_ptr   = 1;
 128     *(AVFrame *)data = ractx->frame;
 129
 130     return FRAMESIZE;
 131 }
 132
 133 AVCodec ff_ra_144_decoder = {
 134     .name           = "real_144",
 135     .type           = AVMEDIA_TYPE_AUDIO,
 136     .id             = AV_CODEC_ID_RA_144,
 137     .priv_data_size = sizeof(RA144Context),
 138     .init           = ra144_decode_init,
 139     .decode         = ra144_decode_frame,
 140     .capabilities   = CODEC_CAP_DR1,
 141     .long_name      = NULL_IF_CONFIG_SMALL("RealAudio 1.0 (14.4K)"),
 142 };