Module wavefunction 

Structs

DensityMatrix

Density matrix for mixed states.

PhaseColorMap

Map complex phase to HSV color.

WaveFunctionRenderer1D

Render 1D wave function as glyph brightness (|psi|^2) and phase (hue).

WaveFunctionRenderer2D

Render 2D wave function as glyph brightness grid.

Functions

coherent_state

Generate a coherent state of the harmonic oscillator.

gaussian_wavepacket

Generate a Gaussian wave packet centered at x0 with momentum k0.

momentum_space

DFT to momentum representation.

plane_wave

Generate a plane wave with wave number k.

wigner_function

Compute the Wigner quasi-probability distribution.