def test_nlse_spectrum():
"""Check for reasonable agreement of NLSE output with old PyNLO results."""
pulse = lf.Pulse(pulse_type='sech',
fwhm_ps=0.050,
time_window_ps=7,
npts=2**12,
center_wavelength_nm=1550.0,
epp=50e-12)
fiber1 = lf.Fiber(length=8e-3,
center_wl_nm=1550.0,
gamma_W_m=1,
dispersion=(-0.12, 0, 5e-6))
results = lf.NLSE(pulse,
fiber1,
raman=True,
shock=True,
nsaves=100,
atol=1e-5,
rtol=1e-5,
integrator='lsoda',
print_status=False)
z, f, t, AW, AT = results.get_results()
dB = 10 * np.log10(np.abs(AW[-1])**2)
path = os.path.split(os.path.realpath(__file__))[0]
f_prev, dB_prev = np.loadtxt(path + '/nlse_output.txt',
delimiter=',',
unpack=True,
skiprows=1)
# this is probably overly stringent because we are on a dB scale
np.testing.assert_allclose(dB, dB_prev, rtol=1e-4, atol=0)
# test that the wavelengths function works:
z, new_wls, t, AW_wls, AT = results.get_results_wavelength()
assert np.all(np.isfinite(AW_wls))
power_is_avg=False,
epp=EPP)
# create the fiber!
fiber1 = lf.Fiber(Length * 1e-3,
center_wl_nm=pulseWL,
dispersion=(beta2 * 1e-3, beta3 * 1e-3, beta4 * 1e-3),
gamma_W_m=Gamma * 1e-3,
loss_dB_per_m=Alpha * 100)
t_start = time.time()
results = lf.NLSE(pulse,
fiber1,
raman=Raman,
shock=Steep,
nsaves=Steps,
atol=1e-5,
rtol=1e-5,
integrator='lsoda',
reload_fiber=False)
z, f, t, AW, AT = results.get_results()
t_nlse = time.time() - t_start
z = z * 1e3 # convert to mm
IW_dB = 10 * np.log10(np.abs(AW)**2)
IT_dB = 10 * np.log10(np.abs(AT)**2)
# run the PyNLO method
t_start = time.time()
pulse_pynlo = pynlo.light.DerivedPulses.SechPulse(power=1,
"""Runs a simple example of NLSE pulse propagation."""
import laserfun as lf
p = lf.Pulse(pulse_type='sech', fwhm_ps=0.05, epp=50e-12,
center_wavelength_nm=1550, time_window_ps=7)
f = lf.Fiber(length=0.010, center_wl_nm=1550, dispersion=(-0.12, 0, 5e-6),
gamma_W_m=1)
results = lf.NLSE(p, f, print_status=False)
if __name__ == '__main__': # make plots if we're not running tests
results.plot()
p = lf.Pulse(pulse_type='sech',
power=10000,
npts=2**13,
fwhm_ps=0.0284 * 1.76,
center_wavelength_nm=835,
time_window_ps=12.5)
# betas = [beta2, beta3, ...] in units [ps^2/m, ps^3/m ...]
betas = [
-11.830e-3, 8.1038e-5, -9.5205e-8, 2.0737e-10, -5.3943e-13, 1.3486e-15,
-2.5495e-18, 3.0524e-21, -1.7140e-24
]
f = lf.Fiber(length=0.15, center_wl_nm=835, dispersion=betas, gamma_W_m=0.11)
r = lf.NLSE(p, f, nsaves=200, raman=True)
z, new_wls, t, AW, AT = r.get_results_wavelength(wmin=400,
wmax=1350,
wn=400,
datatype='dB')
fig, axs = plt.subplots(1, 2, figsize=(8, 5), tight_layout='True')
axs[0].imshow(AW,
aspect='auto',
origin='lower',
extent=(new_wls.min(), new_wls.max(), z.min(), z.max()),
clim=(np.max(AW) - 40, np.max(AW)),
cmap='jet')
axs[1].imshow(AT,
beta3 = (1-frac) * beta3i + frac * beta3f
beta4 = (1-frac) * beta4i + frac * beta4f
return (beta2*1e-3, beta3*1e-3, beta4*1e-3)
def gamma_function(z):
"""Set the gamma."""
frac = z/(Length*1e-3)
gamma = 1e-3*((1-frac)*Gammai + frac*Gammaf)
return gamma
def alpha_function(z):
"""Set the loss."""
frac = z/(Length*1e-3)
alpha = ((1-frac)*Alphai + frac*Alphaf)
return alpha*100
fiber1.set_dispersion_function(dispersion_function)
fiber1.set_gamma_function(gamma_function)
fiber1.set_alpha_function(alpha_function)
# propagate the pulse using the NLSE
results = lf.NLSE(pulse, fiber1, raman=Raman, shock=Steep, nsaves=Steps,
atol=atol, rtol=rtol, integrator='lsoda', reload_fiber=True,
print_status=False)
if __name__ == '__main__':
results.plot()
pulse = lf.Pulse(pulse_type='sech',
fwhm_ps=FWHM,
center_wavelength_nm=pulseWL,
time_window_ps=Window,
GDD=GDD,
TOD=TOD,
npts=Points,
power_is_avg=False,
epp=EPP)
# create the fiber!
fiber1 = lf.Fiber(Length * 1e-3,
center_wl_nm=pulseWL,
dispersion_format='GVD',
dispersion=(beta2 * 1e-3, beta3 * 1e-3, beta4 * 1e-3),
gamma_W_m=Gamma * 1e-3,
loss_dB_per_m=Alpha * 100)
# propagate the pulse using the NLSE
results = lf.NLSE(pulse,
fiber1,
raman=Raman,
shock=Steep,
nsaves=Steps,
rtol=rtol,
atol=atol,
print_status=False)
if __name__ == '__main__':
results.plot()