def get_odgauss(): n = 2 l = 10 grid_length = 20 num_grid_points = 1001 weight = 1 center = 0 deviation = 2.5 odgauss = GeneralOrbitalSystem( n, ODQD( l, grid_length, num_grid_points, potential=ODQD.GaussianPotential(weight, center, deviation, np=np), ), ) return odgauss
n = 2 l = 20 grid_length = 5 num_grid_points = 1001 omega = 1 odho = ODQD(n, l, grid_length, num_grid_points) odho.setup_system(potential=ODQD.HOPotential(omega)) save_data(odho, "odho") length_of_dw = 5 oddw = ODQD(n, l, 6, num_grid_points) oddw.setup_system(potential=ODQD.DWPotential(omega, length_of_dw)) save_data(oddw, "oddw") weight = 1 center = 0 deviation = 2.5 odgauss = ODQD(n, l, 20, num_grid_points) odgauss.setup_system( potential=ODQD.GaussianPotential(weight, center, deviation, np=np)) save_data(odgauss, "odgauss") oddw_smooth = ODQD(n, l, grid_length, num_grid_points) oddw_smooth.setup_system(potential=ODQD.DWPotentialSmooth(a=5)) save_data(oddw_smooth, "oddw_smooth")
odho.setup_system(potential=ODQD.HOPotential(omega)) plot_one_body_system(odho) length_of_dw = 5 dw = ODQD(n, l, 6, num_grid_points) dw.setup_system(potential=ODQD.DWPotential(omega, length_of_dw)) plot_one_body_system(dw) weight = 1 center = 0 deviation = 2.5 gauss = ODQD(n, l, 20, num_grid_points) gauss.setup_system( potential=ODQD.GaussianPotential(weight, center, deviation, np=np)) plot_one_body_system(gauss) tw_gauss_potential = lambda x: ( ODQD.GaussianPotential(weight, center, deviation, np) (x) + ODQD.GaussianPotential(weight, center - 10, deviation, np) (x) + ODQD.GaussianPotential(weight, center + 10, deviation, np)(x)) tw_gauss = ODQD(n, l, 30, num_grid_points) tw_gauss.setup_system(potential=tw_gauss_potential) plot_one_body_system(tw_gauss) dw_smooth = ODQD(n, l, grid_length, num_grid_points) dw_smooth.setup_system(potential=ODQD.DWPotentialSmooth(a=5)) plot_one_body_system(dw_smooth)