82 lines
2.8 KiB
Python
82 lines
2.8 KiB
Python
|
"""① 块长一对一验证 ② 数字输出底噪测量。"""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
from scipy.io import wavfile
|
||
|
|
|
||
|
|
PROJ = "/home/lou/桌面/工作区/实验/collaplex音效"
|
||
|
|
sys.path.insert(0, os.path.join(PROJ, "dsp"))
|
||
|
|
|
||
|
|
import common # noqa: E402
|
||
|
|
|
||
|
|
PY = os.path.join(PROJ, ".venv/bin/python")
|
||
|
|
EQ = os.path.join(PROJ, "dsp/eq.py")
|
||
|
|
REC = "/tmp/noise_probe.wav"
|
||
|
|
|
||
|
|
|
||
|
|
def find_digital_sink() -> str:
|
||
|
|
raw = subprocess.run(["pw-dump"], capture_output=True, text=True).stdout
|
||
|
|
for node in json.loads(raw):
|
||
|
|
props = (node.get("info") or {}).get("props") or {}
|
||
|
|
name = str(props.get("node.name", ""))
|
||
|
|
if props.get("media.class") == "Audio/Sink" and "iec958" in name:
|
||
|
|
return name
|
||
|
|
return ""
|
||
|
|
|
||
|
|
|
||
|
|
def block_test() -> None:
|
||
|
|
print("=== ① 块长一对一(不等长块不应被填充) ===")
|
||
|
|
store = common.open_store()
|
||
|
|
for nbytes in (4096, 2048, 1024, 512, 384):
|
||
|
|
n = nbytes // 4
|
||
|
|
tone = np.sin(2 * np.pi * 1000 * np.arange(n) / 96000.0)
|
||
|
|
x = (0.3 * tone).astype("<f4")
|
||
|
|
common.write_params(store, [0.0] * 32, 0.0)
|
||
|
|
proc = subprocess.run([PY, EQ], input=x.tobytes(), capture_output=True, timeout=60)
|
||
|
|
ok = len(proc.stdout) == len(x.tobytes())
|
||
|
|
print(" 喂 %5d 字节 -> 出 %5d 字节 %s" % (len(x.tobytes()), len(proc.stdout),
|
||
|
|
"✓ 一对一" if ok else "✗ 被填充!"))
|
||
|
|
|
||
|
|
|
||
|
|
def noise_test(sink: str) -> None:
|
||
|
|
print()
|
||
|
|
print("=== ② 数字输出底噪(静默 3 秒, 单位 dBFS) ===")
|
||
|
|
if not sink:
|
||
|
|
print(" 找不到数字输出, 跳过")
|
||
|
|
return
|
||
|
|
for label, extra in (("静默(无播放)", []),):
|
||
|
|
subprocess.run(["timeout", "3", "pw-record", "--target", sink,
|
||
|
|
"-P", "{ stream.capture.sink = true }",
|
||
|
|
"--rate", "96000", "--channels", "2", "--format", "f32", REC, *extra],
|
||
|
|
stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
||
|
|
try:
|
||
|
|
_rate, data = wavfile.read(REC)
|
||
|
|
except (OSError, ValueError):
|
||
|
|
print(" 录音失败")
|
||
|
|
return
|
||
|
|
arr = data.astype(np.float64)
|
||
|
|
if arr.ndim > 1:
|
||
|
|
arr = arr[:, 0]
|
||
|
|
if arr.size == 0:
|
||
|
|
print(" 空录音")
|
||
|
|
return
|
||
|
|
rms = float(np.sqrt(np.mean(arr ** 2)))
|
||
|
|
peak = float(np.max(np.abs(arr)))
|
||
|
|
print(" %-12s RMS %7.2f dBFS | 峰值 %7.2f dBFS | 样点 %d" % (
|
||
|
|
label, 20 * np.log10(max(rms, 1e-12)), 20 * np.log10(max(peak, 1e-12)), arr.size))
|
||
|
|
print(" (参考: 数字域本底一般应低于 -90 dBFS; 明显高于此值说明链路在自造噪声)")
|
||
|
|
|
||
|
|
|
||
|
|
def main() -> None:
|
||
|
|
block_test()
|
||
|
|
noise_test(find_digital_sink())
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|