import numpy as np
import trap_units as si
from scipy import constants as const

waist_trap_r = 135 * const.micro  # waist of the trap in radial direction
waist_trap_z = 17 * const.micro  # waist of the trap in axial direction
waist_rep = 100 * const.micro  # waist of the repulsion beam
omega_trap_r = 150 * 2 * np.pi  # radial trap frequency

grad_r = 0.33 * si.G / si.cm
grad_z = 50 * si.G / si.cm

wavelength_resonance = 671 * const.nano
wavelength_rep = 650 * const.nano
wavelength_trap = 1064 * const.nano
omega_0 = 2 * np.pi * const.c / wavelength_resonance
omega_rep = 2 * np.pi * const.c / wavelength_rep
omega_trap = 2 * np.pi * const.c / wavelength_trap
# Einstein coefficient
a_21 = gamma = 2 * np.pi * 5.8724 * const.mega

# 2d beam angle
theta = np.deg2rad(7.35)

d = wavelength_trap / (2 * waist_trap_z) * waist_trap_r

mass_lithium6 = 6.0151228 * const.value("atomic mass constant")