dedebenui/scgenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

spec_0 and field_0 refactor

Browse Source
This commit is contained in:
Benoît Sierro
2023-01-27 09:18:25 +01:00
parent 5cc817d0d3
commit 391e074cd2
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/scgenerator/evaluator.py
View File

@@ -310,6 +310,8 @@ default_rules: list[Rule] = [
# Pulse # Pulse
Rule("field_0", pulse.finalize_pulse), Rule("field_0", pulse.finalize_pulse),
Rule(["input_time", "input_field"], pulse.load_custom_field), Rule(["input_time", "input_field"], pulse.load_custom_field),
Rule("spec_0", lambda fft, field_0: fft(field_0)),
Rule("field_0", lambda ifft, spec_0: ifft(spec_0)),
Rule("spec_0", utils.load_previous_spectrum, ["recovery_data_dir"], priorities=4), Rule("spec_0", utils.load_previous_spectrum, ["recovery_data_dir"], priorities=4),
Rule("spec_0", utils.load_previous_spectrum, priorities=3), Rule("spec_0", utils.load_previous_spectrum, priorities=3),
*Rule.deduce( *Rule.deduce(
@@ -401,8 +403,6 @@ envelope_rules = default_rules + [
Rule(["w_c", "w", "w_order"], math.build_envelope_w_grid), Rule(["w_c", "w", "w_order"], math.build_envelope_w_grid),
# Pulse # Pulse
Rule("pre_field_0", pulse.initial_field_envelope, priorities=1), Rule("pre_field_0", pulse.initial_field_envelope, priorities=1),
Rule("spec_0", np.fft.fft, ["field_0"]),
Rule("field_0", np.fft.ifft, ["spec_0"]),
Rule("c_to_a_factor", pulse.c_to_a_factor), Rule("c_to_a_factor", pulse.c_to_a_factor),
# Dispersion # Dispersion
Rule(["wl_for_disp", "dispersion_ind"], fiber.lambda_for_envelope_dispersion), Rule(["wl_for_disp", "dispersion_ind"], fiber.lambda_for_envelope_dispersion),
@@ -439,8 +439,6 @@ full_field_rules = default_rules + [
# Grid # Grid
Rule(["w", "w_order", "l"], math.build_full_field_w_grid, priorities=1), Rule(["w", "w_order", "l"], math.build_full_field_w_grid, priorities=1),
# Pulse # Pulse
Rule("spec_0", np.fft.rfft, ["field_0"]),
Rule("field_0", np.fft.irfft, ["spec_0"]),
Rule("pre_field_0", pulse.initial_full_field), Rule("pre_field_0", pulse.initial_full_field),
# Dispersion # Dispersion
Rule(["wl_for_disp", "dispersion_ind"], fiber.lambda_for_full_field_dispersion), Rule(["wl_for_disp", "dispersion_ind"], fiber.lambda_for_full_field_dispersion),