lou
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
"""三点电平测量: 虚拟声卡入口 / HRTF 输入(=归一化输出) / 数字输出。用来定位哪一级没处理。"""
|
|
import array
|
|
import json
|
|
import math
|
|
import subprocess
|
|
import time
|
|
import wave
|
|
|
|
RATE = 48000
|
|
DEVNULL = subprocess.DEVNULL
|
|
SRC = "/tmp/h_white.wav"
|
|
|
|
|
|
def ids() -> dict[str, int]:
|
|
d = json.loads(subprocess.run(["pw-dump"], capture_output=True, text=True).stdout)
|
|
out = {}
|
|
for n in d:
|
|
if n.get("type") != "PipeWire:Interface:Node":
|
|
continue
|
|
p = n.get("info", {}).get("props", {}) or {}
|
|
nm = p.get("node.name") or ""
|
|
if p.get("media.class") == "Audio/Sink":
|
|
if nm == "collaplex_vsink":
|
|
out["vsink"] = n["id"]
|
|
elif nm == "collaplex_hrtf_in":
|
|
out["hrtf_in"] = n["id"]
|
|
elif nm.endswith("iec958-stereo"):
|
|
out["out"] = n["id"]
|
|
elif nm == "alsa_input.usb-EDIFIER_Technology_EDIFIER_Fit900NB_4250315939393214-00.mono-fallback":
|
|
out["mic"] = n["id"]
|
|
return out
|
|
|
|
|
|
def rec(target: int, secs: float, path: str) -> float:
|
|
r = subprocess.Popen(["pw-record", "--target", str(target), "-P", "{ stream.capture.sink = true }",
|
|
"--rate", str(RATE), "--channels", "2", path], stderr=DEVNULL)
|
|
time.sleep(secs)
|
|
r.terminate()
|
|
r.wait()
|
|
w = wave.open(path)
|
|
n = w.getnframes()
|
|
d = array.array("h")
|
|
d.frombytes(w.readframes(n))
|
|
if not d:
|
|
return -99
|
|
rms = (sum(x * x for x in d) / len(d)) ** 0.5 / 32768
|
|
return 20 * math.log10(rms) if rms > 0 else -99
|
|
|
|
|
|
ID = ids()
|
|
print("节点 id: %s" % ID)
|
|
|
|
print("\n--- 静音时(应 -99 = 抓的是 sink monitor; 若 -50 左右 = 抓到麦克风) ---")
|
|
for k in ("vsink", "hrtf_in", "out", "mic"):
|
|
if k in ID:
|
|
print(" %-8s %.2f dBFS" % (k, rec(ID[k], 1.0, "/tmp/t_%s.wav" % k)))
|
|
|
|
print("\n--- 播素材时(素材 -12 dBFS) ---")
|
|
p = subprocess.Popen(["pw-play", "--target", "collaplex_vsink", SRC], stderr=DEVNULL)
|
|
time.sleep(0.5)
|
|
res = {}
|
|
for k in ("vsink", "hrtf_in", "out", "mic"):
|
|
if k in ID:
|
|
res[k] = rec(ID[k], 1.2, "/tmp/p_%s.wav" % k)
|
|
p.terminate()
|
|
p.wait()
|
|
for k, v in res.items():
|
|
print(" %-8s %.2f dBFS" % (k, v))
|