A Wigner-based ray-tracing method for imaging simulations