def yz_shift_general(self, scale, step, z): shift = np.zeros((step+1, step+1), complex) for x in range(0, step+1): for y in range(0, step+1): xp = -scale + 2*scale/step*x yp = -scale + 2*scale/step*y shift[y][x] = hfunc.light_shift(z, xp, yp, self.dipole, self.laserconfig, self.polconfig) return shift.real
def yz_shift_general(self, scale, step, z): shift = np.zeros((step + 1, step + 1), complex) for x in range(0, step + 1): for y in range(0, step + 1): xp = -scale + 2 * scale / step * x yp = -scale + 2 * scale / step * y shift[y][x] = hfunc.light_shift(z, xp, yp, self.dipole, self.laserconfig, self.polconfig) return shift.real
def xz_shift(self, scale, step): global y_min y_min = hfunc.y_opt(self.dipole, self.laserconfig, self.polconfig) shift = np.zeros((step+1, step+1), complex) for x in range(0, step+1): for y in range(0, step+1): xp = -scale + 2*scale/step*x yp = -scale + 2*scale/step*y shift[y][x] = hfunc.light_shift(xp, y_min, yp, self.dipole, self.laserconfig, self.polconfig) return shift.real
def xz_shift(self, scale, step): global y_min y_min = hfunc.y_opt(self.dipole, self.laserconfig, self.polconfig) shift = np.zeros((step + 1, step + 1), complex) for x in range(0, step + 1): for y in range(0, step + 1): xp = -scale + 2 * scale / step * x yp = -scale + 2 * scale / step * y shift[y][x] = hfunc.light_shift(xp, y_min, yp, self.dipole, self.laserconfig, self.polconfig) return shift.real