auracaster/bumble_mirror
3
0
Fork 1
mirror of https://github.com/google/bumble.git synced 2026-05-08 03:58:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix support for float32

Browse Source
This commit is contained in:
Gilles Boccon-Gibod
2025-02-08 18:12:45 -05:00
parent 5f2d24570e
commit 859bb0609f
2 changed files with 19 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
bumble/audio/io.py
View File

@@ -68,7 +68,7 @@ class PcmFormat:
sample_type_str, sample_rate_str, channels_str = format_str.split(',')
if sample_type_str == 'int16le':
sample_type = cls.SampleType.INT16
elif sample_rate_str == 'float32le':
elif sample_type_str == 'float32le':
sample_type = cls.SampleType.FLOAT32
else:
raise ValueError(f'sample type {sample_type_str} not supported')
@@ -77,6 +77,10 @@ class PcmFormat:
return cls(endianness, sample_type, sample_rate, channels)
@property
def bytes_per_sample(self) -> int:
return 2 if self.sample_type == self.SampleType.INT16 else 4
def check_audio_output(output: str) -> bool:
if output == 'device' or output.startswith('device:'):
@@ -486,7 +490,9 @@ class StreamAudioInput(ThreadedAudioInput):
return self._pcm_format
def _read(self, frame_size: int) -> bytes:
return self._stream.read(frame_size * self._pcm_format.channels * 2)
return self._stream.read(
frame_size * self._pcm_format.channels * self._pcm_format.bytes_per_sample
)
class FileAudioInput(StreamAudioInput):
@@ -496,9 +502,6 @@ class FileAudioInput(StreamAudioInput):
self._stream = open(filename, "rb")
super().__init__(self._stream, pcm_format)
def _read(self, frame_size: int) -> bytes:
return self._stream.read(frame_size * self._pcm_format.channels * 2)
def _close(self) -> None:
self._stream.close()