av/cmd/audio-player/adpcm.js

128 lines
3.5 KiB
JavaScript

/*
NAME
adpcm.js
AUTHOR
Trek Hopton <trek@ausocean.org>
LICENSE
This file is Copyright (C) 2018 the Australian Ocean Lab (AusOcean)
It is free software: you can redistribute it and/or modify them
under the terms of the GNU General Public License as published by the
Free Software Foundation, either version 3 of the License, or (at your
option) any later version.
It 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 in gpl.txt.
If not, see [GNU licenses](http://www.gnu.org/licenses).
*/
/*
Original IMA/DVI ADPCM specification: (http://www.cs.columbia.edu/~hgs/audio/dvi/IMA_ADPCM.pdf).
Reference algorithms for ADPCM compression and decompression are in part 6.
*/
const indexTable = [
-1, -1, -1, -1, 2, 4, 6, 8,
-1, -1, -1, -1, 2, 4, 6, 8
]
const stepTable = [
7, 8, 9, 10, 11, 12, 13, 14,
16, 17, 19, 21, 23, 25, 28, 31,
34, 37, 41, 45, 50, 55, 60, 66,
73, 80, 88, 97, 107, 118, 130, 143,
157, 173, 190, 209, 230, 253, 279, 307,
337, 371, 408, 449, 494, 544, 598, 658,
724, 796, 876, 963, 1060, 1166, 1282, 1411,
1552, 1707, 1878, 2066, 2272, 2499, 2749, 3024,
3327, 3660, 4026, 4428, 4871, 5358, 5894, 6484,
7132, 7845, 8630, 9493, 10442, 11487, 12635, 13899,
15289, 16818, 18500, 20350, 22385, 24623, 27086, 29794,
32767
]
const byteDepth = 2, // We are working with 16-bit samples. TODO(Trek): make configurable.
initSamps = 2, // Number of samples used to initialise the encoder.
initBytes = initSamps * byteDepth,
headBytes = 4, // Number of bytes in the header of ADPCM.
samplesPerEnc = 2, // Number of sample encoded at a time eg. 2 16-bit samples get encoded into 1 byte.
bytesPerEnc = samplesPerEnc * byteDepth,
compFact = 4; // In general ADPCM compresses by a factor of 4.
let est = 0, // Estimation of sample based on quantised ADPCM nibble.
idx = 0, // Index to step used for estimation.
step = 0;
// decodeSample takes 4 bits which represents a single ADPCM nibble, and returns a 16 bit decoded PCM sample.
function decodeSample(nibble) {
let diff = 0;
if ((nibble & 4) != 0) {
diff += step;
}
if ((nibble & 2) != 0) {
diff += step >> 1;
}
if ((nibble & 1) != 0) {
diff += step >> 2;
}
diff += step >> 3;
if ((nibble & 8) != 0) {
diff = -diff;
}
est += diff;
idx += indexTable[nibble];
if (idx < 0) {
idx = 0;
} else if (idx > stepTable.length - 1) {
idx = stepTable.length - 1;
}
step = stepTable[idx];
result = est;
return result;
}
// decode takes an array of bytes of arbitrary length representing adpcm and decodes it into pcm.
function decode(b) {
// Initialize Decoder with first 4 bytes of b.
est = bytesToInt16(b[0], b[1]);
idx = b[byteDepth];
step = stepTable[idx];
var result = b.slice(0, 2);
for (var i = headBytes; i < b.length - b[3]; i++) {
var twoNibs = b[i];
var nib2 = twoNibs >> 4;
var nib1 = (nib2 << 4) ^ twoNibs
var sample1 = int16ToBytes(decodeSample(nib1))
result.push(...sample1)
var sample2 = int16ToBytes(decodeSample(nib2))
result.push(...sample2)
}
if (b[3] == 1) {
var padNib = b[b.length - 1]
var sample = int16ToBytes(decodeSample(padNib))
result.push(...sample)
}
return result;
}
function int16ToBytes(num) {
return [(num & 0x00ff), (num & 0xff00) >> 8];
}
function bytesToInt16(b) {
return (b[0] | (b[1] << 8))
}