技能备份 - 2026-04-15 (40个技能)
This commit is contained in:
@@ -0,0 +1,65 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Generate Discord voice message waveform from an ogg file.
|
||||
Usage: gen_waveform.py <ogg_file>
|
||||
Output: JSON with duration_secs and base64-encoded waveform (256 points, 0-255).
|
||||
"""
|
||||
|
||||
import subprocess
|
||||
import struct
|
||||
import base64
|
||||
import json
|
||||
import sys
|
||||
|
||||
def main():
|
||||
if len(sys.argv) < 2:
|
||||
print("Usage: gen_waveform.py <ogg_file>")
|
||||
sys.exit(1)
|
||||
|
||||
ogg_file = sys.argv[1]
|
||||
|
||||
# Get duration
|
||||
result = subprocess.run(
|
||||
["ffprobe", "-v", "error", "-show_entries", "format=duration",
|
||||
"-of", "csv=p=0", ogg_file],
|
||||
capture_output=True, text=True
|
||||
)
|
||||
duration_secs = float(result.stdout.strip())
|
||||
|
||||
# Extract raw PCM (mono, 16-bit, 48kHz)
|
||||
result = subprocess.run(
|
||||
["ffmpeg", "-i", ogg_file, "-f", "s16le", "-ac", "1",
|
||||
"-ar", "48000", "-"],
|
||||
capture_output=True
|
||||
)
|
||||
pcm_data = result.stdout
|
||||
|
||||
# Parse samples
|
||||
num_samples = len(pcm_data) // 2
|
||||
samples = struct.unpack(f"<{num_samples}h", pcm_data[:num_samples * 2])
|
||||
|
||||
# Generate 256-point waveform
|
||||
points = 256
|
||||
chunk_size = max(1, num_samples // points)
|
||||
waveform = []
|
||||
|
||||
for i in range(points):
|
||||
start = i * chunk_size
|
||||
end = min(start + chunk_size, num_samples)
|
||||
if start >= num_samples:
|
||||
waveform.append(0)
|
||||
continue
|
||||
chunk = samples[start:end]
|
||||
avg = sum(abs(s) for s in chunk) / len(chunk)
|
||||
# Map to 0-255
|
||||
val = min(255, int(avg / 32768 * 255))
|
||||
waveform.append(val)
|
||||
|
||||
waveform_b64 = base64.b64encode(bytes(waveform)).decode()
|
||||
|
||||
print(json.dumps({
|
||||
"duration_secs": round(duration_secs, 2),
|
||||
"waveform": waveform_b64
|
||||
}))
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user