Source code for librosa.core.fft

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""Fast Fourier Transform (FFT) library container"""

__all__ = ["get_fftlib", "set_fftlib"]

# Object to hold FFT interfaces
__FFTLIB = None

[docs]def set_fftlib(lib=None): """Set the FFT library used by librosa. Parameters ---------- lib : None or module Must implement an interface compatible with `numpy.fft`. If ``None``, reverts to `numpy.fft`. Examples -------- Use `pyfftw`: >>> import pyfftw >>> librosa.set_fftlib(pyfftw.interfaces.numpy_fft) Reset to default `numpy` implementation >>> librosa.set_fftlib() """ global __FFTLIB if lib is None: from numpy import fft lib = fft __FFTLIB = lib
[docs]def get_fftlib(): """Get the FFT library currently used by librosa Returns ------- fft : module The FFT library currently used by librosa. Must API-compatible with `numpy.fft`. """ global __FFTLIB return __FFTLIB
# Set the FFT library to numpy's, by default set_fftlib(None)