/pyFFTW

A pythonic python wrapper around FFTW

Primary LanguagePythonOtherNOASSERTION

Current Build Status

Travis Appveyor Read the Docs
travis_ci appveyor_ci read_the_docs

Conda-forge Status

Linux OSX Windows

Conda-forge Info

Name Downloads Version Platforms
Conda Recipe Conda Downloads Conda Version Conda Platforms

PyFFTW

pyFFTW is a pythonic wrapper around FFTW 3, the speedy FFT library. The ultimate aim is to present a unified interface for all the possible transforms that FFTW can perform.

Both the complex DFT and the real DFT are supported, as well as on arbitrary axes of abitrary shaped and strided arrays, which makes it almost feature equivalent to standard and real FFT functions of numpy.fft (indeed, it supports the clongdouble dtype which numpy.fft does not).

Wisdom import and export now works fairly reliably.

Operating FFTW in multithreaded mode is supported.

pyFFTW implements the numpy and scipy fft interfaces in order for users to take advantage of the speed of FFTW with minimal code modifications.

A comprehensive unittest suite can be found with the source on the GitHub repository or with the source distribution on PyPI.

The documentation can be found on Read the Docs the source is on GitHub and the python package index page PyPI. Issues and questions can be raised at the GitHub Issues page.

Requirements (i.e. what it was designed for)

  • Python 2.7 or >= 3.4
  • Numpy >= 1.10.4 (lower versions may work)
  • FFTW >= 3.3 (lower versions may work) libraries for single, double, and long double precision in serial and multithreading (pthreads or openMP) versions.
  • Cython >= 0.29

(install these as much as possible with your preferred package manager).

In practice, pyFFTW may work with older versions of these dependencies, but it is not tested against them.

Optional Dependencies

Scipy and Dask are only required in order to use their respective interfaces.

Installation

We recommend not building from github, but using the release on the python package index with tools such as pip:

pip install pyfftw

Pre-built binary wheels for 64-bit Linux, Mac OS X and Windows are available on the PyPI page for all supported Python versions.

Installation from PyPI may also work on other systems when the FFTW libraries are available, but other platforms have not been tested.

Alternatively, users of the conda package manager can install from the conda-forge channel via:

conda install -c conda-forge pyfftw

Windows development builds are also automatically uploaded to bintray as wheels (which are built against numpy 1.10), from where they can be downloaded and installed with something like::

pip install pyFFTW-0.11.1+3.g898bce5-cp36-cp36m-win_amd64.whl

where the version and the revision hash are set accordingly.

Read on if you do want to build from source...

Building

To build in place:

python setup.py build_ext --inplace

or:

pip install -r requirements.txt -e . -v

That cythonizes the python extension and builds it into a shared library which is placed in pyfftw/. The directory can then be treated as a python package.

After you've run setup.py with cython available, you then have a normal C extension in the pyfftw directory. Further building does not depend on cython (as long as the .c file remains).

During configuration the available FFTW libraries are detected, so pay attention to the output when running setup.py. On certain platforms, for example the long double precision is not available. pyFFTW still builds fine but will fail at runtime if asked to perform a transform involving long double precision.

Regarding multithreading, if both posix and openMP FFTW libs are available, the openMP libs are preferred. This preference can be reversed by defining the environment variable PYFFTW_USE_PTHREADS prior to building. If neither option is available, pyFFTW works in serial mode only.

For more ways of building and installing, see the distutils documentation and setuptools documentation.

Platform specific build info

Windows

To build for windows from source, download the fftw dlls for your system and the header file from here (they're in a zip file) and place them in the pyfftw directory. The files are libfftw3-3.dll, libfftw3l-3.dll, libfftw3f-3.dll. These libs use pthreads for multithreading. If you're using a version of FFTW other than 3.3, it may be necessary to copy fftw3.h into include\win.

The builds on PyPI use mingw for the 32-bit release and the Windows SDK C++ compiler for the 64-bit release. The scripts should handle this automatically. If you want to compile for 64-bit Windows, you have to use the MS Visual C++ compiler. Set up your environment as described here and then run setup.py with the version of python you wish to target and a suitable build command.

For using the MS Visual C++ compiler, you'll need to create a set of suitable .lib files as described on the FFTW page.

Mac OSX

Install FFTW from homebrew::

brew install fftw

Set temporary environmental variables, such that pyfftw finds fftw::

export DYLD_LIBRARY_PATH=/usr/local/lib export LDFLAGS="-L/usr/local/lib" export CFLAGS="-I/usr/local/include"

Now install pyfftw from pip::

pip install pyfftw

It has been suggested that macports might also work fine. You should then replace the LD environmental variables above with the right ones.

  • DYLD - path for libfftw3.dylib etc - find /usr -name libfftw3.dylib
  • LDFLAGS - path for fftw3.h - find /usr -name fftw3.h

FreeBSD

Install FFTW from ports tree or pkg:

- math/fftw3
- math/fftw3-float
- math/fftw3-long

Please install all of them, if possible.

Contributions

Contributions are always welcome and valued. The primary restriction on accepting contributions is that they are exhaustively tested. The bulk of pyFFTW has been developed in a test-driven way (i.e. the test to be satisfied is written before the code). I strongly encourage potential contributors to adopt such an approach.

See some of my philosophy on testing in development [here] (https://hgomersall.wordpress.com/2014/10/03/from-test-driven-development-and-specifications). If you want to argue with the philosophy, there is probably a good place to do it.

New contributions should adhere to PEP 8, but this is only weakly enforced (there is loads of legacy stuff that breaks it, and things like a single trailing whitespace is not a big deal).

The best place to start with contributing is by raising an issue detailing the specifics of what you wish to achieve (there should be a clear use-case for any new functionality). I tend to respond pretty quickly and am happy to help where I can with any conceptual issues.

I suggest reading the issues already open in order that you know where things might be heading, or what others are working on.