mirror of
https://github.com/HiMeditator/auto-caption.git
synced 2026-02-04 04:14:42 +08:00
eba2c5ca45
- 重构字幕引擎，将音频采集改为在新线程上进行 - 重构 audio2text 中的类，调整运行逻辑 - 更新 main 函数，添加对 Sosv 模型的支持 - 修改 AudioStream 类，默认使用 16000Hz 采样率
110 lines
3.0 KiB
Python
110 lines
3.0 KiB
Python
"""获取 Linux 系统音频输入流"""
|
||
|
||
import subprocess
|
||
from textwrap import dedent
|
||
|
||
|
||
def find_monitor_source():
|
||
result = subprocess.run(
|
||
["pactl", "list", "short", "sources"],
|
||
stdout=subprocess.PIPE, text=True
|
||
)
|
||
lines = result.stdout.splitlines()
|
||
|
||
for line in lines:
|
||
parts = line.split('\t')
|
||
if len(parts) >= 2 and ".monitor" in parts[1]:
|
||
return parts[1]
|
||
|
||
raise RuntimeError("System output monitor device not found")
|
||
|
||
|
||
def find_input_source():
|
||
result = subprocess.run(
|
||
["pactl", "list", "short", "sources"],
|
||
stdout=subprocess.PIPE, text=True
|
||
)
|
||
lines = result.stdout.splitlines()
|
||
|
||
for line in lines:
|
||
parts = line.split('\t')
|
||
name = parts[1]
|
||
if ".monitor" not in name:
|
||
return name
|
||
|
||
raise RuntimeError("Microphone input device not found")
|
||
|
||
|
||
class AudioStream:
|
||
"""
|
||
获取系统音频流
|
||
|
||
初始化参数：
|
||
audio_type: 0-系统音频输出流（不支持，不会生效），1-系统音频输入流（默认）
|
||
chunk_rate: 每秒采集音频块的数量，默认为10
|
||
"""
|
||
def __init__(self, audio_type=1, chunk_rate=10):
|
||
self.audio_type = audio_type
|
||
|
||
if self.audio_type == 0:
|
||
self.source = find_monitor_source()
|
||
else:
|
||
self.source = find_input_source()
|
||
self.stop_signal = False
|
||
self.process = None
|
||
self.FORMAT = 16
|
||
self.SAMP_WIDTH = 2
|
||
self.CHANNELS = 2
|
||
self.RATE = 16000
|
||
self.CHUNK_RATE = chunk_rate
|
||
self.CHUNK = self.RATE // chunk_rate
|
||
|
||
def get_info(self):
|
||
dev_info = f"""
|
||
音频捕获进程：
|
||
- 捕获类型：{"音频输出" if self.audio_type == 0 else "音频输入"}
|
||
- 设备源：{self.source}
|
||
- 捕获进程 PID：{self.process.pid if self.process else "None"}
|
||
|
||
样本格式：{self.FORMAT}
|
||
样本位宽：{self.SAMP_WIDTH}
|
||
样本通道数：{self.CHANNELS}
|
||
样本采样率：{self.RATE}
|
||
样本块大小：{self.CHUNK}
|
||
"""
|
||
print(dev_info)
|
||
|
||
def open_stream(self):
|
||
"""
|
||
启动音频捕获进程
|
||
"""
|
||
self.process = subprocess.Popen(
|
||
["parec", "-d", self.source, "--format=s16le", "--rate=16000", "--channels=2"],
|
||
stdout=subprocess.PIPE
|
||
)
|
||
|
||
def read_chunk(self):
|
||
"""
|
||
读取音频数据
|
||
"""
|
||
if self.stop_signal:
|
||
self.close_stream()
|
||
return None
|
||
if self.process and self.process.stdout:
|
||
return self.process.stdout.read(self.CHUNK)
|
||
return None
|
||
|
||
def close_stream_signal(self):
|
||
"""
|
||
线程安全的关闭系统音频输入流，不一定会立即关闭
|
||
"""
|
||
self.stop_signal = True
|
||
|
||
def close_stream(self):
|
||
"""
|
||
关闭系统音频捕获进程
|
||
"""
|
||
if self.process:
|
||
self.process.terminate()
|
||
self.stop_signal = False
|