danvics/Handy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(audio): prioritize F32 sample format for better quality (#390) (#393)

Browse source
This commit is contained in:
mg-dev25 2025-11-30 09:59:54 +01:00 committed by GitHub
parent d52bec9d87
commit 244a99e623
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 22 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
src-tauri/src/audio_toolkit/audio/recorder.rs
View file

@ -204,19 +204,36 @@ impl AudioRecorder {
device: &cpal::Device,
) -> Result<cpal::SupportedStreamConfig, Box<dyn std::error::Error>> {
let supported_configs = device.supported_input_configs()?;
let mut best_config: Option<cpal::SupportedStreamConfigRange> = None;
// Try to find a config that supports 16kHz
// Try to find a config that supports 16kHz, prioritizing better formats
for config_range in supported_configs {
if config_range.min_sample_rate().0 <= constants::WHISPER_SAMPLE_RATE
&& config_range.max_sample_rate().0 >= constants::WHISPER_SAMPLE_RATE
{
// Found a config that supports 16kHz, use it
return Ok(
config_range.with_sample_rate(cpal::SampleRate(constants::WHISPER_SAMPLE_RATE))
);
match best_config {
None => best_config = Some(config_range),
Some(ref current) => {
// Prioritize F32 > I16 > I32 > others
let score = |fmt: cpal::SampleFormat| match fmt {
cpal::SampleFormat::F32 => 4,
cpal::SampleFormat::I16 => 3,
cpal::SampleFormat::I32 => 2,
_ => 1,
};
if score(config_range.sample_format()) > score(current.sample_format()) {
best_config = Some(config_range);
}
}
}
}
}
if let Some(config) = best_config {
return Ok(config.with_sample_rate(cpal::SampleRate(constants::WHISPER_SAMPLE_RATE)));
}
// If no config supports 16kHz, fall back to default
Ok(device.default_input_config()?)
}