waveq

Waveq is a simulation for the 2D Schrodinger equation with arbitrary potential terms. It is based on taichi and python. To do this, you need python3.10 and no later and install taichi by

pip install taichi

once this is done, you can run

python wavefunction.py

with the following controls:

[SPACE]: change view modes between probability and real/imaginary parts
[ESC]: exit
P: Overlay a visualization of the potential
C: Clear, i.e. set the wavefunction to zero
[MOUSE]: Clicking (and/or dragging) the mouse disturbs the wavefunction by a pulse. 

Here are some pictures from this