libavcodec/dpcm.c

   1 /*
   2  * Assorted DPCM codecs
   3  * Copyright (c) 2003 The ffmpeg Project.
   4  *
   5  * This file is part of FFmpeg.
   6  *
   7  * FFmpeg is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * FFmpeg is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with FFmpeg; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21
  22 /**
  23  * @file: dpcm.c
  24  * Assorted DPCM (differential pulse code modulation) audio codecs
  25  * by Mike Melanson (melanson@pcisys.net)
  26  * Xan DPCM decoder by Mario Brito (mbrito@student.dei.uc.pt)
  27  * for more information on the specific data formats, visit:
  28  *   http://www.pcisys.net/~melanson/codecs/simpleaudio.html
  29  * SOL DPCMs implemented by Konstantin Shishkov
  30  *
  31  * Note about using the Xan DPCM decoder: Xan DPCM is used in AVI files
  32  * found in the Wing Commander IV computer game. These AVI files contain
  33  * WAVEFORMAT headers which report the audio format as 0x01: raw PCM.
  34  * Clearly incorrect. To detect Xan DPCM, you will probably have to
  35  * special-case your AVI demuxer to use Xan DPCM if the file uses 'Xxan'
  36  * (Xan video) for its video codec. Alternately, such AVI files also contain
  37  * the fourcc 'Axan' in the 'auds' chunk of the AVI header.
  38  */
  39
  40 #include "avcodec.h"
  41
  42 typedef struct DPCMContext {
  43     int channels;
  44     short roq_square_array[256];
  45     long sample[2];//for SOL_DPCM
  46     const int *sol_table;//for SOL_DPCM
  47 } DPCMContext;
  48
  49 #define SE_16BIT(x)  if (x & 0x8000) x -= 0x10000;
  50
  51 static int interplay_delta_table[] = {
  52          0,      1,      2,      3,      4,      5,      6,      7,
  53          8,      9,     10,     11,     12,     13,     14,     15,
  54         16,     17,     18,     19,     20,     21,     22,     23,
  55         24,     25,     26,     27,     28,     29,     30,     31,
  56         32,     33,     34,     35,     36,     37,     38,     39,
  57         40,     41,     42,     43,     47,     51,     56,     61,
  58         66,     72,     79,     86,     94,    102,    112,    122,
  59        133,    145,    158,    173,    189,    206,    225,    245,
  60        267,    292,    318,    348,    379,    414,    452,    493,
  61        538,    587,    640,    699,    763,    832,    908,    991,
  62       1081,   1180,   1288,   1405,   1534,   1673,   1826,   1993,
  63       2175,   2373,   2590,   2826,   3084,   3365,   3672,   4008,
  64       4373,   4772,   5208,   5683,   6202,   6767,   7385,   8059,
  65       8794,   9597,  10472,  11428,  12471,  13609,  14851,  16206,
  66      17685,  19298,  21060,  22981,  25078,  27367,  29864,  32589,
  67     -29973, -26728, -23186, -19322, -15105, -10503,  -5481,     -1,
  68          1,      1,   5481,  10503,  15105,  19322,  23186,  26728,
  69      29973, -32589, -29864, -27367, -25078, -22981, -21060, -19298,
  70     -17685, -16206, -14851, -13609, -12471, -11428, -10472,  -9597,
  71      -8794,  -8059,  -7385,  -6767,  -6202,  -5683,  -5208,  -4772,
  72      -4373,  -4008,  -3672,  -3365,  -3084,  -2826,  -2590,  -2373,
  73      -2175,  -1993,  -1826,  -1673,  -1534,  -1405,  -1288,  -1180,
  74      -1081,   -991,   -908,   -832,   -763,   -699,   -640,   -587,
  75       -538,   -493,   -452,   -414,   -379,   -348,   -318,   -292,
  76       -267,   -245,   -225,   -206,   -189,   -173,   -158,   -145,
  77       -133,   -122,   -112,   -102,    -94,    -86,    -79,    -72,
  78        -66,    -61,    -56,    -51,    -47,    -43,    -42,    -41,
  79        -40,    -39,    -38,    -37,    -36,    -35,    -34,    -33,
  80        -32,    -31,    -30,    -29,    -28,    -27,    -26,    -25,
  81        -24,    -23,    -22,    -21,    -20,    -19,    -18,    -17,
  82        -16,    -15,    -14,    -13,    -12,    -11,    -10,     -9,
  83         -8,     -7,     -6,     -5,     -4,     -3,     -2,     -1
  84
  85 };
  86
  87 static const int sol_table_old[16] =
  88     { 0x0,  0x1,  0x2 , 0x3,  0x6,  0xA,  0xF, 0x15,
  89     -0x15, -0xF, -0xA, -0x6, -0x3, -0x2, -0x1, 0x0};
  90
  91 static const int sol_table_new[16] =
  92     { 0x0,  0x1,  0x2,  0x3,  0x6,  0xA,  0xF,  0x15,
  93       0x0, -0x1, -0x2, -0x3, -0x6, -0xA, -0xF, -0x15};
  94
  95 static const int sol_table_16[128] = {
  96     0x000, 0x008, 0x010, 0x020, 0x030, 0x040, 0x050, 0x060, 0x070, 0x080,
  97     0x090, 0x0A0, 0x0B0, 0x0C0, 0x0D0, 0x0E0, 0x0F0, 0x100, 0x110, 0x120,
  98     0x130, 0x140, 0x150, 0x160, 0x170, 0x180, 0x190, 0x1A0, 0x1B0, 0x1C0,
  99     0x1D0, 0x1E0, 0x1F0, 0x200, 0x208, 0x210, 0x218, 0x220, 0x228, 0x230,
 100     0x238, 0x240, 0x248, 0x250, 0x258, 0x260, 0x268, 0x270, 0x278, 0x280,
 101     0x288, 0x290, 0x298, 0x2A0, 0x2A8, 0x2B0, 0x2B8, 0x2C0, 0x2C8, 0x2D0,
 102     0x2D8, 0x2E0, 0x2E8, 0x2F0, 0x2F8, 0x300, 0x308, 0x310, 0x318, 0x320,
 103     0x328, 0x330, 0x338, 0x340, 0x348, 0x350, 0x358, 0x360, 0x368, 0x370,
 104     0x378, 0x380, 0x388, 0x390, 0x398, 0x3A0, 0x3A8, 0x3B0, 0x3B8, 0x3C0,
 105     0x3C8, 0x3D0, 0x3D8, 0x3E0, 0x3E8, 0x3F0, 0x3F8, 0x400, 0x440, 0x480,
 106     0x4C0, 0x500, 0x540, 0x580, 0x5C0, 0x600, 0x640, 0x680, 0x6C0, 0x700,
 107     0x740, 0x780, 0x7C0, 0x800, 0x900, 0xA00, 0xB00, 0xC00, 0xD00, 0xE00,
 108     0xF00, 0x1000, 0x1400, 0x1800, 0x1C00, 0x2000, 0x3000, 0x4000
 109 };
 110
 111
 112
 113 static av_cold int dpcm_decode_init(AVCodecContext *avctx)
 114 {
 115     DPCMContext *s = avctx->priv_data;
 116     int i;
 117     short square;
 118
 119     s->channels = avctx->channels;
 120     s->sample[0] = s->sample[1] = 0;
 121
 122     switch(avctx->codec->id) {
 123
 124     case CODEC_ID_ROQ_DPCM:
 125         /* initialize square table */
 126         for (i = 0; i < 128; i++) {
 127             square = i * i;
 128             s->roq_square_array[i] = square;
 129             s->roq_square_array[i + 128] = -square;
 130         }
 131         break;
 132
 133
 134     case CODEC_ID_SOL_DPCM:
 135         switch(avctx->codec_tag){
 136         case 1:
 137             s->sol_table=sol_table_old;
 138             s->sample[0] = s->sample[1] = 0x80;
 139             break;
 140         case 2:
 141             s->sol_table=sol_table_new;
 142             s->sample[0] = s->sample[1] = 0x80;
 143             break;
 144         case 3:
 145             s->sol_table=sol_table_16;
 146             break;
 147         default:
 148             av_log(avctx, AV_LOG_ERROR, "Unknown SOL subcodec\n");
 149             return -1;
 150         }
 151         break;
 152
 153     default:
 154         break;
 155     }
 156
 157     return 0;
 158 }
 159
 160 static int dpcm_decode_frame(AVCodecContext *avctx,
 161                              void *data, int *data_size,
 162                              const uint8_t *buf, int buf_size)
 163 {
 164     DPCMContext *s = avctx->priv_data;
 165     int in, out = 0;
 166     int predictor[2];
 167     int channel_number = 0;
 168     short *output_samples = data;
 169     int shift[2];
 170     unsigned char byte;
 171     short diff;
 172
 173     if (!buf_size)
 174         return 0;
 175
 176     // almost every DPCM variant expands one byte of data into two
 177     if(*data_size/2 < buf_size)
 178         return -1;
 179
 180     switch(avctx->codec->id) {
 181
 182     case CODEC_ID_ROQ_DPCM:
 183         if (s->channels == 1)
 184             predictor[0] = AV_RL16(&buf[6]);
 185         else {
 186             predictor[0] = buf[7] << 8;
 187             predictor[1] = buf[6] << 8;
 188         }
 189         SE_16BIT(predictor[0]);
 190         SE_16BIT(predictor[1]);
 191
 192         /* decode the samples */
 193         for (in = 8, out = 0; in < buf_size; in++, out++) {
 194             predictor[channel_number] += s->roq_square_array[buf[in]];
 195             predictor[channel_number] = av_clip_int16(predictor[channel_number]);
 196             output_samples[out] = predictor[channel_number];
 197
 198             /* toggle channel */
 199             channel_number ^= s->channels - 1;
 200         }
 201         break;
 202
 203     case CODEC_ID_INTERPLAY_DPCM:
 204         in = 6;  /* skip over the stream mask and stream length */
 205         predictor[0] = AV_RL16(&buf[in]);
 206         in += 2;
 207         SE_16BIT(predictor[0])
 208         output_samples[out++] = predictor[0];
 209         if (s->channels == 2) {
 210             predictor[1] = AV_RL16(&buf[in]);
 211             in += 2;
 212             SE_16BIT(predictor[1])
 213             output_samples[out++] = predictor[1];
 214         }
 215
 216         while (in < buf_size) {
 217             predictor[channel_number] += interplay_delta_table[buf[in++]];
 218             predictor[channel_number] = av_clip_int16(predictor[channel_number]);
 219             output_samples[out++] = predictor[channel_number];
 220
 221             /* toggle channel */
 222             channel_number ^= s->channels - 1;
 223         }
 224
 225         break;
 226
 227     case CODEC_ID_XAN_DPCM:
 228         in = 0;
 229         shift[0] = shift[1] = 4;
 230         predictor[0] = AV_RL16(&buf[in]);
 231         in += 2;
 232         SE_16BIT(predictor[0]);
 233         if (s->channels == 2) {
 234             predictor[1] = AV_RL16(&buf[in]);
 235             in += 2;
 236             SE_16BIT(predictor[1]);
 237         }
 238
 239         while (in < buf_size) {
 240             byte = buf[in++];
 241             diff = (byte & 0xFC) << 8;
 242             if ((byte & 0x03) == 3)
 243                 shift[channel_number]++;
 244             else
 245                 shift[channel_number] -= (2 * (byte & 3));
 246             /* saturate the shifter to a lower limit of 0 */
 247             if (shift[channel_number] < 0)
 248                 shift[channel_number] = 0;
 249
 250             diff >>= shift[channel_number];
 251             predictor[channel_number] += diff;
 252
 253             predictor[channel_number] = av_clip_int16(predictor[channel_number]);
 254             output_samples[out++] = predictor[channel_number];
 255
 256             /* toggle channel */
 257             channel_number ^= s->channels - 1;
 258         }
 259         break;
 260     case CODEC_ID_SOL_DPCM:
 261         in = 0;
 262         if (avctx->codec_tag != 3) {
 263             if(*data_size/4 < buf_size)
 264                 return -1;
 265             while (in < buf_size) {
 266                 int n1, n2;
 267                 n1 = (buf[in] >> 4) & 0xF;
 268                 n2 = buf[in++] & 0xF;
 269                 s->sample[0] += s->sol_table[n1];
 270                  if (s->sample[0] < 0) s->sample[0] = 0;
 271                 if (s->sample[0] > 255) s->sample[0] = 255;
 272                 output_samples[out++] = (s->sample[0] - 128) << 8;
 273                 s->sample[s->channels - 1] += s->sol_table[n2];
 274                 if (s->sample[s->channels - 1] < 0) s->sample[s->channels - 1] = 0;
 275                 if (s->sample[s->channels - 1] > 255) s->sample[s->channels - 1] = 255;
 276                 output_samples[out++] = (s->sample[s->channels - 1] - 128) << 8;
 277             }
 278         } else {
 279             while (in < buf_size) {
 280                 int n;
 281                 n = buf[in++];
 282                 if (n & 0x80) s->sample[channel_number] -= s->sol_table[n & 0x7F];
 283                 else s->sample[channel_number] += s->sol_table[n & 0x7F];
 284                 s->sample[channel_number] = av_clip_int16(s->sample[channel_number]);
 285                 output_samples[out++] = s->sample[channel_number];
 286                 /* toggle channel */
 287                 channel_number ^= s->channels - 1;
 288             }
 289         }
 290         break;
 291     }
 292
 293     *data_size = out * sizeof(short);
 294     return buf_size;
 295 }
 296
 297 #define DPCM_DECODER(id, name)                  \
 298 AVCodec name ## _decoder = {                    \
 299     #name,                                      \
 300     CODEC_TYPE_AUDIO,                           \
 301     id,                                         \
 302     sizeof(DPCMContext),                        \
 303     dpcm_decode_init,                           \
 304     NULL,                                       \
 305     NULL,                                       \
 306     dpcm_decode_frame,                          \
 307 };
 308
 309 DPCM_DECODER(CODEC_ID_INTERPLAY_DPCM, interplay_dpcm);
 310 DPCM_DECODER(CODEC_ID_ROQ_DPCM, roq_dpcm);
 311 DPCM_DECODER(CODEC_ID_SOL_DPCM, sol_dpcm);
 312 DPCM_DECODER(CODEC_ID_XAN_DPCM, xan_dpcm);