Crates.io | rubato |
lib.rs | rubato |
version | 0.16.1 |
source | src |
created_at | 2020-05-06 18:34:51.681739 |
updated_at | 2024-10-27 16:02:06.440001 |
description | Asynchronous resampling library intended for audio data |
homepage | |
repository | https://github.com/HEnquist/rubato |
max_upload_size | |
id | 238271 |
size | 317,553 |
An audio sample rate conversion library for Rust.
This library provides resamplers to process audio in chunks.
The ratio between input and output sample rates is completely free. Implementations are available that accept a fixed length input while returning a variable length output, and vice versa.
Rubato can be used in realtime applications without any allocation during processing by preallocating a [Resampler] and using its input_buffer_allocate and output_buffer_allocate methods before beginning processing. The log feature feature should be disabled for realtime use (it is disabled by default).
Input and output data are stored in a non-interleaved format.
Input and output data are stored as slices of references, &[AsRef<[f32]>]
or &[AsRef<[f64]>]
.
The inner references (AsRef<[f32]>
or AsRef<[f64]>
) hold the sample values for one channel each.
Since normal vectors implement the AsRef
trait,
Vec<Vec<f32>>
and Vec<Vec<f64>>
can be used for both input and output.
The asynchronous resamplers are available with and without anti-aliasing filters.
Resampling with anti-aliasing is based on band-limited interpolation using sinc interpolation filters. The sinc interpolation upsamples by an adjustable factor, and then the new sample points are calculated by interpolating between these points. The resampling ratio can be updated at any time.
Resampling without anti-aliasing omits the cpu-heavy sinc interpolation. This runs much faster but produces a lower quality result.
Synchronous resampling is implemented via FFT. The data is FFT:ed, the spectrum modified, and then inverse FFT:ed to get the resampled data. This type of resampler is considerably faster but doesn't support changing the resampling ratio.
The resamplers provided by this library are intended to process audio in chunks. The optimal chunk size is determined by the application, but will likely end up somwhere between a few hundred to a few thousand frames. This gives a good compromize between efficiency and memory usage.
Rubato is suitable for real-time applications when using the Resampler::process_into_buffer()
method.
This stores the output in a pre-allocated output buffer, and performs no allocations or other
operations that may block the thread.
A suggested simple process for resampling an audio clip of known length to a new sample rate is as follows. Here it is assumed that the source data is stored in a vec, or some other structure that supports reading arbitrary number of frames at a time. For simplicity, the output is stored in a temporary buffer during resampling, and copied to the destination afterwards.
Preparations:
Resampler::output_delay()
to know how many frames of delay the resampler gives.
Store the number as delay
.new_length = original_length * new_rate / original_rate
.Now it's time to process the bulk of the clip by repeated procesing calls. Loop:
Resampler::input_frames_next()
to learn how many frames the resampler needs.Resampler::process()
or Resampler::process_into_buffer()
.The next step is to process the last remaining frames.
Resampler::process_partial()
or Resampler::process_partial_into_buffer()
.At this point, all frames have been sent to the resampler,
but because of the delay through the resampler,
it may still have some frames in its internal buffers.
When all wanted frames have been generated, the length of the temporary
output buffer should be at least new_length + delay
.
If this is not the case, call Resampler::process_partial()
or Resampler::process_partial_into_buffer()
with None
as input,
and append the output to the temporary output buffer.
If needed, repeat until the length is sufficient.
Finally, copy the data from the temporary output buffer to the desired destination.
Skip the first delay
frames, and copy new_length
frames.
If there is more than one clip to resample from and to the same sample rates,
the same resampler should be reused.
Creating a new resampler is an expensive task and should be avoided if possible.
Start the procedire from the start, but instead of creating a new resampler,
call Resampler::reset()
on the existing one to prepare it for a new job.
When resamping a stream, the process is normally performed in real time, and either the input of output is some API that provides or consumes frames at a given rate.
Audio APIs such as CoreAudio on MacOS, or the cross platform cpal crate, often use callback functions for data exchange.
A complete
When capturing audio from these, the application passes a function to the audio API.
The API then calls this function periodically, with a pointer to a data buffer containing new audio frames.
The data buffer size is usually the same on every call, but that varies between APIs.
It is important that the function does not block,
since this would block some internal loop of the API and cause loss of some audio data.
It is recommended to keep the callback function light.
Ideally it should read the provided audio data from the buffer provided by the API,
and optionally perform some light processing such as sample format conversion.
No heavy processing such as resampling should be performed here.
It should then store the audio data to a shared buffer.
The buffer may be a Arc<Mutex<VecDeque<T>>>
,
or something more advanced such as ringbuf.
A separate loop, running either in the main or a separate thread, should then read from that buffer, resample, and save to file. If the Audio API provides a fixed buffer size, then this number of frames is a good choice for the resampler chunk size. If the size varies, the shared buffer can be used to adapt the chunk sizes of the audio API and the resampler. A good starting point for the resampler chunk size is to use an "easy" value near the average chunk size of the audio API. Make sure that the shared buffer is large enough to not get full in case for the loop gets blocked waiting for example for disk access.
The loop should follow a process similar to resampling a clip, but the input is now the shared buffer. The loop needs to wait for the needed number of frames to become available in the buffer, before reading and passing them to the resampler.
It would also be appropriate to omit the temporary output buffer, and write the output directly to the destination. The hound crate is a popular choice for reading and writing uncompressed audio formats.
The asynchronous resampler supports SIMD on x86_64 and on aarch64. The SIMD capabilities of the CPU are determined at runtime. If no supported SIMD instruction set is available, it falls back to a scalar implementation.
On x86_64, it will try to use AVX. If AVX isn't available, it will instead try SSE3.
On aarch64 (64-bit Arm), it will use Neon if available.
The synchronous resamplers benefit from the SIMD support of the RustFFT library.
fft_resampler
: Enable the FFT based synchronous resamplersThis feature is enabled by default. Disable it if the FFT resamplers are not needed, to save compile time and reduce the resulting binary size.
log
: Enable loggingThis feature enables logging via the log
crate. This is intended for debugging purposes.
Note that outputting logs allocates a [std::string::String] and most logging implementations involve various other system calls.
These calls may take some (unpredictable) time to return, during which the application is blocked.
This means that logging should be avoided if using this library in a realtime application.
The log
feature can be enabled when running tests, which can be very useful when debugging.
The logging level can be set via the RUST_LOG
environment variable.
Example:
RUST_LOG=trace cargo test --features log
Resample a single chunk of a dummy audio file from 44100 to 48000 Hz. See also the "process_f64" example that can be used to process a file from disk.
use rubato::{Resampler, SincFixedIn, SincInterpolationType, SincInterpolationParameters, WindowFunction};
let params = SincInterpolationParameters {
sinc_len: 256,
f_cutoff: 0.95,
interpolation: SincInterpolationType::Linear,
oversampling_factor: 256,
window: WindowFunction::BlackmanHarris2,
};
let mut resampler = SincFixedIn::<f64>::new(
48000 as f64 / 44100 as f64,
2.0,
params,
1024,
2,
).unwrap();
let waves_in = vec![vec![0.0f64; 1024];2];
let waves_out = resampler.process(&waves_in, None).unwrap();
The examples
directory contains a few sample applications for testing the resamplers.
There are also Python scripts for generating simple test signals as well as analyzing the resampled results.
The examples read and write raw audio data in 64-bit float format.
They can be used to process .wav files if the files are first converted to the right format.
Use sox
to convert a .wav to raw samples:
sox some_file.wav -e floating-point -b 64 some_file_f64.raw
After processing, the result can be converted back to new .wav. This examples converts to 16-bits at 44.1 kHz:
sox -e floating-point -b 64 -r 44100 -c 2 resampler_output.raw -e signed-integer -b 16 some_file_resampled.wav
Many audio editors, for example Audacity, are also able to directly import and export the raw samples.
The rubato
crate requires rustc version 1.61 or newer.
fft_resampler
feature.log
feature.input/output_buffer_allocate()
optionally pre-fill buffers with zeros.Accept any AsRef<[T]> as input.
License: MIT