Add trace param to animate_wigner

[tjstavenger-pnnl]

As in plot_wigner, add support to first perform partial trace on each frame's state vector.