71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
"""WEM encoding/decoding: setup transcoding, hash/alloc formulas, encode -> decode fidelity."""
|
|
import numpy as np
|
|
import pytest
|
|
|
|
from ackit import audio, encoder
|
|
from ackit.allocsize import decode_alloc, fnv1_32
|
|
from ackit.convert import library
|
|
from ackit.known_setups import db
|
|
from ackit.wem import Wem
|
|
from ackit.wwvorbis import transcode_setup, W2V, V2W
|
|
import synth
|
|
|
|
|
|
def test_codebook_library_is_the_game_one():
|
|
assert len(library().books) == 598
|
|
|
|
|
|
@pytest.mark.parametrize("entry", db())
|
|
def test_hash_and_alloc_formulas_reproduce_every_game_setup(entry):
|
|
s = bytes.fromhex(entry["setup"])
|
|
ch = int(list(entry["formats"])[0].split("/")[1][:-2])
|
|
assert fnv1_32(s) == entry["hash"]
|
|
assert decode_alloc(s, ch, library())[0] == entry["alloc"]
|
|
|
|
|
|
@pytest.mark.parametrize("rate,ch", [(24000, 1), (28000, 1), (44100, 2), (32000, 6)])
|
|
def test_encode_decode_roundtrip(rate, ch):
|
|
x, sr = synth.tone(1.5, 48000, 440, 1)
|
|
wem, info = encoder.encode(x, sr, rate, ch)
|
|
m = Wem.parse(wem)
|
|
assert m.channels == ch and m.rate == info["rate"]
|
|
# setup survives a Wwise -> Vorbis -> Wwise round trip
|
|
vs, _ = transcode_setup(m.setup_packet(), W2V, ch, library())
|
|
assert transcode_setup(vs, V2W, ch, library())[0] == m.setup_packet()
|
|
y, ysr = audio.decode_wem(wem)
|
|
ref = audio.resample(x, sr, ysr)[:, 0]
|
|
n = min(len(ref), len(y))
|
|
assert abs(len(y) - len(ref)) < 64
|
|
assert np.corrcoef(ref[:n], y[:n, 0])[0, 1] > 0.98
|
|
|
|
|
|
def test_six_channels_at_48k_fall_back_to_32k():
|
|
x, sr = synth.tone(0.5, 48000, 440, 1)
|
|
_, info = encoder.encode(x, sr, 48000, 6)
|
|
assert info["rate"] == 32000
|
|
|
|
|
|
def test_encode_fit_shrinks_or_fails():
|
|
x, sr = synth.tone(3.0, 24000, 440, 1)
|
|
full, _ = encoder.encode(x, sr, 24000, 1)
|
|
wem, info = encoder.encode_fit(x, sr, 24000, 1, limit=int(len(full) * 0.8))
|
|
assert len(wem) <= len(full) * 0.8 and "shrunk_from" in info
|
|
with pytest.raises(encoder.EncodeError):
|
|
encoder.encode_fit(x, sr, 24000, 1, limit=300)
|
|
|
|
|
|
def test_audio_helpers():
|
|
x, sr = synth.tone(1.0, 48000, 440, 2)
|
|
assert audio.to_channels(x, 1).shape == (48000, 1)
|
|
assert audio.to_channels(x[:, :1], 6).shape == (48000, 6)
|
|
assert abs(audio.peak_db(audio.normalize(x, -3.0)) + 3.0) < 0.01
|
|
assert audio.trim(x, sr, 0.25, 0.75).shape[0] == 24000
|
|
assert audio.resample(x, 48000, 24000).shape[0] == 24000
|
|
|
|
|
|
def test_long_multichannel_encode():
|
|
"""libsndfile's Vorbis writer overflowed the stack on one huge write (30 s x 6 ch); we write in blocks."""
|
|
x, sr = synth.tone(30.0, 32000, 330, 6)
|
|
wem, info = encoder.encode(x, sr, 32000, 6)
|
|
assert Wem.parse(wem).vorb["total_frames"] == 30 * 32000
|