Spaces:
Running
Running
File size: 3,247 Bytes
c441d2c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 | import sounddevice as sd
import threading
import queue
from typing import Union, Optional, Callable
import numpy as np
import os
import soundfile as sf
def run_in_sub_thread(func) -> Callable[..., threading.Thread]:
def wrapper(*args, **kwargs) -> threading.Thread:
thread = threading.Thread(target=func, args=args, kwargs=kwargs)
thread.daemon = True
thread.start()
return thread
return wrapper
class AudioPlayer:
CHUNK_SIZE: int = 1024
def __init__(self):
self._task_queue: queue.Queue[bytes | str] = queue.Queue()
self._worker_thread: Optional[threading.Thread] = None
self._stop_event: threading.Event = threading.Event()
self._start_worker()
def _start_worker(self):
"""启动工作线程（如果它尚未运行或已关闭）。"""
if self._worker_thread and self._worker_thread.is_alive():
return
self._stop_event.clear()
self._worker_thread = self._playback_worker()
@run_in_sub_thread
def _playback_worker(self) -> None:
while not self._stop_event.is_set():
try:
task: str = self._task_queue.get(timeout=0.1)
except queue.Empty:
continue
stream = None
try:
if isinstance(task, str) and os.path.isfile(task):
with sf.SoundFile(task, 'r') as f:
if sd is not None:
stream = sd.OutputStream(
samplerate=f.samplerate,
channels=f.channels,
dtype='float32',
)
stream.start()
while not self._stop_event.is_set():
chunk = f.read(self.CHUNK_SIZE, dtype='float32')
if not chunk.any():
break
stream.write(chunk)
except Exception as e:
if isinstance(e, sf.SoundFileError):
print(f"无法读取或解析音频文件: {task}, 错误: {e}")
else:
print(f"播放时发生错误: {e}")
finally:
if stream:
stream.stop()
stream.close()
self._task_queue.task_done()
def play(self, source: Union[str, np.ndarray]):
"""将音频源加入播放队列。"""
self._start_worker()
self._task_queue.put(source)
def stop(self):
"""停止播放并清空播放队列。"""
self._stop_event.set()
if self._worker_thread and self._worker_thread.is_alive():
self._worker_thread.join()
self._stop_event.clear()
with self._task_queue.mutex:
self._task_queue.queue.clear()
while self._task_queue.unfinished_tasks > 0:
self._task_queue.task_done()
def wait(self):
"""阻塞，直到队列中所有任务都播放完成。"""
self._task_queue.join()
def close(self):
"""永久关闭播放器并释放资源。"""
self.stop()
|