pde_opt.numerics.utils.initialization_utils

This module contains the initialization functions for the PDEs.

Functions

add_vortex_to_wavefunction(psi, vortex_pos)

Add a vortex to an existing wavefunction at a specified position

initialize_Psi(N[, width, vortexnumber])

Function for creating an initial condition for GPE
pde_opt.numerics.utils.initialization_utils.initialize_Psi(N, width=100, vortexnumber=0)[source]

Function for creating an initial condition for GPE

Parameters:

  • N – number of grid points

  • width – width of blob

  • vortextnumber – number of vortices to intialize

Returns:

initial condition for GPE simulation

pde_opt.numerics.utils.initialization_utils.add_vortex_to_wavefunction(psi, vortex_pos, vortex_strength=1, vortex_width=1)[source]

Add a vortex to an existing wavefunction at a specified position

Parameters:

  • psi – existing wavefunction (complex array)

  • vortex_pos – (x, y) position of the vortex center

  • vortex_strength – winding number of the vortex (default=1)

  • vortex_width – width of the vortex core (default=10)

Returns:

modified wavefunction with vortex added