mirror of https://github.com/AxioDL/amuse.git
134 lines
4.3 KiB
C++
134 lines
4.3 KiB
C++
#include "amuse/N64MusyXCodec.hpp"
|
|
|
|
#include <cstdint>
|
|
#include <cstring>
|
|
|
|
/* Acknowledgements:
|
|
* SubDrag for N64 Sound Tool (http://www.goldeneyevault.com/viewfile.php?id=212)
|
|
* Bobby Smiles for MusyX codec research
|
|
*/
|
|
|
|
static int rdot(unsigned n, const int16_t* x, const int16_t* y) {
|
|
int accu = 0;
|
|
|
|
y += n;
|
|
|
|
while (n != 0) {
|
|
accu += ((int)*(x++) * (int)*(--y));
|
|
--n;
|
|
}
|
|
|
|
return accu;
|
|
}
|
|
|
|
static int16_t adpcm_get_predicted_sample(unsigned char byte, unsigned char mask, unsigned lshift, unsigned rshift) {
|
|
int16_t sample = ((int16_t)byte & (int16_t)mask) << lshift;
|
|
sample >>= rshift; /* signed */
|
|
return sample;
|
|
}
|
|
|
|
static void adpcm_get_predicted_frame(int16_t* dst, const unsigned char* src, const unsigned char* nibbles,
|
|
unsigned rshift) {
|
|
*(dst++) = (src[0] << 8) | src[1];
|
|
*(dst++) = (src[2] << 8) | src[3];
|
|
|
|
for (unsigned i = 1; i < 16; ++i) {
|
|
unsigned char byteData = nibbles[i];
|
|
|
|
*(dst++) = adpcm_get_predicted_sample(byteData, 0xf0, 8, rshift);
|
|
*(dst++) = adpcm_get_predicted_sample(byteData, 0x0f, 12, rshift);
|
|
}
|
|
}
|
|
|
|
static unsigned adpcm_decode_upto_8_samples(int16_t* dst, const int16_t* src, const int16_t* cb_entry,
|
|
const int16_t* last_samples, unsigned size) {
|
|
if (size == 0)
|
|
return 0;
|
|
|
|
const int16_t* book1 = cb_entry;
|
|
const int16_t* book2 = cb_entry + 8;
|
|
|
|
int16_t l1 = last_samples[0];
|
|
int16_t l2 = last_samples[1];
|
|
|
|
int accu;
|
|
|
|
for (unsigned i = 0; i < size; ++i) {
|
|
accu = (int)src[i] << 11;
|
|
accu += book1[i] * l1 + book2[i] * l2 + rdot(i, book2, src);
|
|
dst[i] = N64MusyXSampClamp(accu >> 11);
|
|
}
|
|
|
|
return size;
|
|
}
|
|
|
|
unsigned N64MusyXDecompressFrame(int16_t* out, const uint8_t* in, const int16_t coefs[8][2][8], unsigned lastSample) {
|
|
int16_t frame[32];
|
|
|
|
unsigned remSamples = lastSample;
|
|
unsigned samples = 0;
|
|
unsigned procSamples, procRes;
|
|
|
|
unsigned char c2 = in[0x8];
|
|
c2 = c2 % 0x80;
|
|
|
|
const int16_t* bookPredictors = coefs[(c2 & 0xf0) >> 4][0];
|
|
unsigned int rshift = (c2 & 0x0f);
|
|
|
|
adpcm_get_predicted_frame(frame, &in[0x0], &in[0x8], rshift);
|
|
|
|
procSamples = (remSamples < 2) ? remSamples : 2;
|
|
memmove(out, frame, 2 * procSamples);
|
|
samples += procSamples;
|
|
remSamples -= procSamples;
|
|
if (samples == lastSample)
|
|
return samples;
|
|
|
|
#define adpcm_decode_upto_end_samples(inOffset, outOffset, size) \
|
|
procSamples = (remSamples < size) ? remSamples : size; \
|
|
procRes = \
|
|
adpcm_decode_upto_8_samples(out + inOffset, frame + inOffset, bookPredictors, out + outOffset, procSamples); \
|
|
samples += procRes; \
|
|
remSamples -= procRes; \
|
|
if (samples == lastSample) \
|
|
return samples;
|
|
|
|
adpcm_decode_upto_end_samples(2, 0, 6);
|
|
adpcm_decode_upto_end_samples(8, 6, 8);
|
|
adpcm_decode_upto_end_samples(16, 14, 8);
|
|
adpcm_decode_upto_end_samples(24, 22, 8);
|
|
|
|
out += 32;
|
|
|
|
c2 = in[0x18];
|
|
c2 = c2 % 0x80;
|
|
|
|
bookPredictors = coefs[(c2 & 0xf0) >> 4][0];
|
|
rshift = (c2 & 0x0f);
|
|
|
|
adpcm_get_predicted_frame(frame, &in[0x4], &in[0x18], rshift);
|
|
|
|
procSamples = (remSamples < 2) ? remSamples : 2;
|
|
memmove(out, frame, 2 * procSamples);
|
|
samples += procSamples;
|
|
remSamples -= procSamples;
|
|
if (samples == lastSample)
|
|
return samples;
|
|
|
|
adpcm_decode_upto_end_samples(2, 0, 6);
|
|
adpcm_decode_upto_end_samples(8, 6, 8);
|
|
adpcm_decode_upto_end_samples(16, 14, 8);
|
|
adpcm_decode_upto_end_samples(24, 22, 8);
|
|
|
|
return samples;
|
|
}
|
|
|
|
unsigned N64MusyXDecompressFrameRanged(int16_t* out, const uint8_t* in, const int16_t coefs[8][2][8],
|
|
unsigned firstSample, unsigned lastSample) {
|
|
int16_t final[64];
|
|
unsigned procSamples = N64MusyXDecompressFrame(final, in, coefs, firstSample + lastSample);
|
|
unsigned samples = procSamples - firstSample;
|
|
memmove(out, final + firstSample, samples * 2);
|
|
return samples;
|
|
}
|