/pyFFTW

A pythonic python wrapper around FFTW

Primary LanguagePythonOtherNOASSERTION

https://travis-ci.org/hgomersall/pyFFTW.png?branch=master

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 github pages, the source is on github and the python package index page is here.

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

  • Python 2.7 or greater (Python 3 is supported)
  • Numpy 1.6
  • FFTW 3.3 or higher (lower versions may work)
  • Cython 0.15 or higher (though the source release on PyPI loses this dependency)

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

Installation

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

pip install pyfftw

or:

easy_install pyfftw

Installers are on the PyPI page for both 32- and 64-bit Windows, which include all the necessary DLLs.

With FFTW installed, the PyPI release should install fine on Linux and Mac OSX. It doesn't mean it won't work anywhere else, just we don't have any information on it.

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

Building

To build in place:

python setup.py build_ext --inplace

That cythons 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).

For more ways of building and installing, see the distutils 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. 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/include"
export CFLAGS="-I/usr/local/include"

Now install pyfftw from pip:

pip install pyfftw

Notes: pkgin fftw package does not contain the long or float implementations of fftw and so installation will fail.

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

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. 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.