mmfutils.math.wigner

Wigner Ville distribution.

This module contains some FFT-based routines for computing the Wigner-Ville distribution.

mmfutils.math.wigner.wigner_ville(psi, dt=1, make_analytic=False, skip=1, pad=True)[source]

Return (ws, P) where P is the Wigner Ville quasi-distribution for psi.

Assumes that psi is periodic. Note: the frequencies at which P is valid are half the frequencies normally associated with the wavefunction. Thus we also return the associated frequencies to avoid possible confusion.

Parameters:

  • psi (array-like) – The input signal.

  • dt (float) – Step size for the input abscissa.

  • make_analytic (bool) – If True, then negative frequency components are set to zero.

  • skip (int) – Downsample the time-domain by skipping this many points.

  • pad (bool) – If True, then pad the input array to remove aliasing artifacts.