import optcom as oc
# Create 2 Gaussian channels
pulse = oc.Gaussian(channels=2,
center_lambda=[1030., 1550.],
peak_power=[0.5, 1.0])
# Create fiber with a user-defined attenuation coefficient
fiber = oc.Fiber(length=1.0,
alpha=[0.4],
ATT=True,
DISP=True,
SPM=True,
save_all=True)
# Create an optical layout and link the first port of 'pulse' to the first port of 'fiber'
layout = oc.Layout()
layout.add_link(pulse.get_port(0), fiber.get_port(0))
layout.run_all()
# Extract outputs and plot
time = fiber.storage.time
power = oc.temporal_power(fiber.storage.channels)
space = fiber.storage.space
oc.animation2d(time,
power,
space,
x_label='t',
y_label='P_t',
plot_title='My first Optcom example',
line_labels=['1030. nm channel', '1550. nm channel'])
#filename='example_anim_readme.mp4')
return output_ports, output_fields
if __name__ == "__main__":
"""Give an example of LoadField usage.
This piece of code is standalone, i.e. can be used in a separate
file as an example.
"""
from typing import List
import numpy as np
import optcom as oc
lt: oc.Layout = oc.Layout()
gssn_1: oc.Gaussian = oc.Gaussian(channels=1, width=[5.0],
field_name='field 1 to be saved in file')
field_saver_1: oc.SaveField = oc.SaveField()
gssn_2: oc.Gaussian = oc.Gaussian(channels=1, width=[10.0],
field_name='field 2 to be saved in file')
field_saver_2: oc.SaveField = oc.SaveField()
lt.add_links((gssn_1[0], field_saver_1[0]), (gssn_2[0], field_saver_2[0]))
lt.run(gssn_1, gssn_2)
fields: List[oc.Field] = field_saver_1.fields + field_saver_2.fields
lt_: oc.Layout = oc.Layout()
load_field: oc.LoadField = oc.LoadField(fields=fields)
oc.plot2d(lambdas,
tf,
x_labels=['nu'],
y_labels=['Amplitude (a.u.)'],
plot_titles=[
"Transfer function centered at "
"{} nm with bandwidth {} nm".format(round(center_lambda, 2),
round(lambda_bw))
])
# Apply on pulse and plot
bit_width = 1000.
domain = oc.Domain(samples_per_bit=2**13,
bit_width=bit_width,
noise_samples=int(1e3),
noise_range=(center_lambda - 1.0, center_lambda + 1.0))
lt: oc.Layout = oc.Layout(domain)
lambda_bw = 0.05 # nm
nu_bw = oc.lambda_bw_to_nu_bw(lambda_bw, center_lambda)
pulse: oc.Gaussian = oc.Gaussian(channels=2,
peak_power=[10.0, 19.0],
width=[10., 6.],
center_lambda=[center_lambda],
noise=100 * np.ones(domain.noise_samples))
filter: oc.GaussianFilter = oc.GaussianFilter(nu_bw=nu_bw,
nu_offset=0.,
order=1,
center_nu=center_nu)
lt.add_link(pulse[0], filter[0])
lt.run(pulse)
plot_titles: List[str] = [
"Original pulse", r"After Gaussian filter with "
from typing import Callable, List, Optional
import numpy as np
import optcom as oc
pulse_1: oc.Gaussian = oc.Gaussian(peak_power=[1.0],
center_lambda=[1550.0])
pulse_2: oc.Gaussian = oc.Gaussian(peak_power=[5.0],
center_lambda=[1030.0])
pulse_3: oc.Gaussian = oc.Gaussian(peak_power=[10.0],
center_lambda=[976.0])
# Dummy component to be able to test the not combining case
pm: oc.IdealPhaseMod = oc.IdealPhaseMod()
lt: oc.Layout = oc.Layout()
combiner: oc.IdealCombiner = oc.IdealCombiner(arms=3, combine=False)
lt.add_links((pulse_1[0], combiner[0]), (pulse_2[0], combiner[1]),
(pulse_3[0], combiner[2]), (combiner[3], pm[0]))
lt.run(pulse_1, pulse_2, pulse_3)
plot_titles: List[str] = ([
"Original pulses", "Pulses coming out of the "
"ideal coupler \n without combination"
])
plot_groups: List[int] = [0, 0, 0, 1]
line_labels: List[Optional[str]] = ['port 0', 'port 1', 'port 2', None]
return output_ports, output_fields
if __name__ == "__main__":
"""Give an example of Sech usage.
This piece of code is standalone, i.e. can be used in a separate
file as an example.
"""
from typing import Callable, List, Optional
import numpy as np
import optcom as oc
lt: oc.Layout = oc.Layout(oc.Domain(samples_per_bit=4096))
channels: int = 3
center_lambda: List[float] = [1552.0, 1549.0, 1596.0]
position: List[float] = [0.3, 0.5]
width: List[float] = [5.3, 6.]
peak_power: List[float] = [1e-3, 2e-3, 6e-3]
rep_freq: List[float] = [0.03, 0.04]
offset_nu: List[float] = [1.56, -1.6]
chirp: List[float] = [0.5, 0.1]
init_phi: List[float] = [1.0, 0.0]
sech = oc.Sech(channels=channels,
center_lambda=center_lambda,
position=position,
fwhm=width,
if __name__ == "__main__":
"""Give an example of CW usage.
This piece of code is standalone, i.e. can be used in a separate
file as an example.
"""
from typing import List
import numpy as np
import optcom as oc
dm: oc.Domain = oc.Domain(samples_per_bit=512, bit_width=100.0)
lt: oc.Layout = oc.Layout(dm)
channels: int = 1
center_lambda: List[float] = [1552.0, 1549.0, 1596.0]
peak_power: List[float] = [1e-3, 2e-3, 6e-3]
offset_nu: List[float] = [0.0, 1.56, -1.6]
init_phi: List[float] = [1.0, 1.0, 0.0]
cw: oc.CW = oc.CW(channels=channels,
center_lambda=center_lambda,
peak_power=peak_power,
offset_nu=offset_nu,
init_phi=init_phi,
save=True)
lt.run(cw)
if __name__ == "__main__":
"""Give an example of FiberCoupler usage.
This piece of code is standalone, i.e. can be used in a separate
file as an example.
"""
import math
from typing import Callable, List, Optional, Union
import numpy as np
import optcom as oc
noise_samples: int = 100
lt: oc.Layout = oc.Layout(
oc.Domain(bit_width=20.0,
samples_per_bit=1024,
noise_samples=noise_samples))
Lambda: float = 1030.0
pulse_1: oc.Gaussian = oc.Gaussian(channels=1,
peak_power=[38.5, 0.5],
fwhm=[1.],
center_lambda=[Lambda],
noise=np.ones(noise_samples) * 12)
pulse_2: oc.Gaussian = oc.Gaussian(channels=1,
peak_power=[23.5, 0.3],
fwhm=[1.],
center_lambda=[1050.0],
noise=np.ones(noise_samples) * 5)
steps: int = int(100)
import numpy as np
import optcom as oc
plot_groups: List[int] = []
line_labels: List[Optional[str]] = []
plot_titles: List[str] = []
x_datas: List[np.ndarray] = []
y_datas: List[np.ndarray] = []
nlse_methods: List[str] = ["ssfm", "ssfm_reduced", "ssfm_symmetric",
"ssfm_opti_reduced", "ssfm_super_sym",
"ssfm_opti_super_sym", "rk4ip", "rk4ip"]
# ---------------- NLSE solvers test -------------------------------
lt: oc.Layout = oc.Layout(oc.Domain(bit_width=100.0, samples_per_bit=4096))
pulse: oc.Gaussian = oc.Gaussian(channels=2, peak_power=[0.5, 1.0], width=[0.5, 0.8])
steps: int = int(5e3)
fiber: oc.Fiber
SS: bool = True
for j, nlse_method in enumerate(nlse_methods):
if (j == len(nlse_methods)-2): # To compute rk4ip and rk4ip_gnlse
oc.set_rk4ip_opti_gnlse(False) # Can make slighty diff. output
else:
oc.set_rk4ip_opti_gnlse(True)
# Propagation
fiber = oc.Fiber(length=0.2, nlse_method=nlse_method, alpha=[0.5],
beta_order=3, gamma=4.0, nl_approx=False, SPM=True,
XPM=True, SS=True, RS=True, steps=steps, save=True)
