Function wigner_function 

pub fn wigner_function(
    psi: &[Complex],
    x_grid: &[f64],
    p_grid: &[f64],
    dx: f64,
    hbar: f64,
) -> Vec<Vec<f64>>

Compute the Wigner quasi-probability distribution.