mirror of
https://github.com/HiMeditator/auto-caption.git
synced 2026-02-04 04:14:42 +08:00
42237a229c
- 新增了系统音频输入（麦克风）的选择功能 - 重构了字幕引擎的构建流程，使用 PyInstaller 打包为可执行文件 - 优化了字幕引擎的启动和停止逻辑 - 更新了用户界面，增加了音频选择的控制选项 - 修改了相关的文件路径和构建配置
127 lines
4.2 KiB
Python
127 lines
4.2 KiB
Python
"""获取 Windows 系统音频输出流"""
|
||
|
||
import pyaudiowpatch as pyaudio
|
||
import numpy as np
|
||
|
||
|
||
def getDefaultLoopbackDevice(mic: pyaudio.PyAudio, info = True)->dict:
|
||
"""
|
||
获取默认的系统音频输出的回环设备
|
||
Args:
|
||
mic (pyaudio.PyAudio): pyaudio对象
|
||
info (bool, optional): 是否打印设备信息
|
||
|
||
Returns:
|
||
dict: 系统音频输出的回环设备
|
||
"""
|
||
try:
|
||
WASAPI_info = mic.get_host_api_info_by_type(pyaudio.paWASAPI)
|
||
except OSError:
|
||
print("Looks like WASAPI is not available on the system. Exiting...")
|
||
exit()
|
||
|
||
default_speaker = mic.get_device_info_by_index(WASAPI_info["defaultOutputDevice"])
|
||
if(info): print("wasapi_info:\n", WASAPI_info, "\n")
|
||
if(info): print("default_speaker:\n", default_speaker, "\n")
|
||
|
||
if not default_speaker["isLoopbackDevice"]:
|
||
for loopback in mic.get_loopback_device_info_generator():
|
||
if default_speaker["name"] in loopback["name"]:
|
||
default_speaker = loopback
|
||
if(info): print("Using loopback device:\n", default_speaker, "\n")
|
||
break
|
||
else:
|
||
print("Default loopback output device not found.")
|
||
print("Run `python -m pyaudiowpatch` to check available devices.")
|
||
print("Exiting...")
|
||
exit()
|
||
|
||
if(info): print(f"Output Stream Device: #{default_speaker['index']} {default_speaker['name']}")
|
||
return default_speaker
|
||
|
||
|
||
def mergeStreamChannels(data, channels):
|
||
"""
|
||
将当前多通道流数据合并为单通道流数据
|
||
|
||
Args:
|
||
data: 多通道数据
|
||
channels: 通道数
|
||
|
||
Returns:
|
||
mono_data_bytes: 单通道数据
|
||
"""
|
||
# (length * channels,)
|
||
data_np = np.frombuffer(data, dtype=np.int16)
|
||
# (length, channels)
|
||
data_np_r = data_np.reshape(-1, channels)
|
||
# (length,)
|
||
mono_data = np.mean(data_np_r.astype(np.float32), axis=1)
|
||
mono_data = mono_data.astype(np.int16)
|
||
mono_data_bytes = mono_data.tobytes()
|
||
return mono_data_bytes
|
||
|
||
class AudioStream:
|
||
"""
|
||
获取系统音频流
|
||
|
||
参数：
|
||
audio_type: （默认）0-系统音频输出流，1-系统音频输入流
|
||
"""
|
||
def __init__(self, audio_type=0):
|
||
self.audio_type = audio_type
|
||
self.mic = pyaudio.PyAudio()
|
||
if self.audio_type == 0:
|
||
self.device = getDefaultLoopbackDevice(self.mic, False)
|
||
else:
|
||
self.device = self.mic.get_default_input_device_info()
|
||
self.stream = None
|
||
self.SAMP_WIDTH = pyaudio.get_sample_size(pyaudio.paInt16)
|
||
self.FORMAT = pyaudio.paInt16
|
||
self.CHANNELS = self.device["maxInputChannels"]
|
||
self.RATE = int(self.device["defaultSampleRate"])
|
||
self.CHUNK = self.RATE // 20
|
||
self.INDEX = self.device["index"]
|
||
|
||
def printInfo(self):
|
||
dev_info = f"""
|
||
采样设备：
|
||
- 设备类型：{ "音频输入" if self.audio_type == 0 else "音频输出" }
|
||
- 序号：{self.device['index']}
|
||
- 名称：{self.device['name']}
|
||
- 最大输入通道数：{self.device['maxInputChannels']}
|
||
- 默认低输入延迟：{self.device['defaultLowInputLatency']}s
|
||
- 默认高输入延迟：{self.device['defaultHighInputLatency']}s
|
||
- 默认采样率：{self.device['defaultSampleRate']}Hz
|
||
- 是否回环设备：{self.device['isLoopbackDevice']}
|
||
|
||
音频样本块大小：{self.CHUNK}
|
||
样本位宽：{self.SAMP_WIDTH}
|
||
音频数据格式：{self.FORMAT}
|
||
音频通道数：{self.CHANNELS}
|
||
音频采样率：{self.RATE}
|
||
"""
|
||
print(dev_info)
|
||
|
||
def openStream(self):
|
||
"""
|
||
打开并返回系统音频输出流
|
||
"""
|
||
if self.stream: return self.stream
|
||
self.stream = self.mic.open(
|
||
format = self.FORMAT,
|
||
channels = self.CHANNELS,
|
||
rate = self.RATE,
|
||
input = True,
|
||
input_device_index = self.INDEX
|
||
)
|
||
return self.stream
|
||
|
||
def closeStream(self):
|
||
"""
|
||
关闭系统音频输出流
|
||
"""
|
||
if self.stream is None: return
|
||
self.stream.stop_stream()
|
||
self.stream.close()
|
||
self.stream = None |