From 39c7ec952d7fff239112f48121c5d087f5f92540 Mon Sep 17 00:00:00 2001 From: Valentin Bruch <valentin.bruch@rwth-aachen.de> Date: Fri, 4 Nov 2022 12:52:12 +0100 Subject: [PATCH] gen_pulse_data.py: back to legacy implementation for old python version on cluster --- gen_pulse_data.py | 65 +++++++++++++++++++++++------------------------ 1 file changed, 32 insertions(+), 33 deletions(-) diff --git a/gen_pulse_data.py b/gen_pulse_data.py index bf188e7..3fca9e9 100644 --- a/gen_pulse_data.py +++ b/gen_pulse_data.py @@ -164,39 +164,38 @@ def main(): ) pulse_shape = options.pop("pulse_shape") - match pulse_shape: - case "gauss": - vdc, fourier_coef = fourier_coef_gauss( - omega = options["omega"], - baseline = options.pop("baseline_voltage"), - duration = options.pop("pulse_duration"), - height = options.pop("pulse_height"), - phase = options.pop("pulse_phase"), - nmax = options["nmax"], - ) - options.pop("pulse_separation") - case "gauss_symmetric": - vdc, fourier_coef = fourier_coef_gauss_symmetric( - omega = options["omega"], - baseline = options.pop("baseline_voltage"), - duration = options.pop("pulse_duration"), - height = options.pop("pulse_height"), - phase = options.pop("pulse_phase"), - nmax = options["nmax"], - ) - options.pop("pulse_separation") - case "updown_gauss": - vdc, fourier_coef = fourier_coef_updown_gauss( - omega = options["omega"], - baseline = options.pop("baseline_voltage"), - duration = options.pop("pulse_duration"), - height = options.pop("pulse_height"), - phase = options.pop("pulse_phase"), - separation = options.pop("pulse_separation"), - nmax = options["nmax"], - ) - case _: - raise ValueError("invalid pulse shape") + if pulse_shape == "gauss": + vdc, fourier_coef = fourier_coef_gauss( + omega = options["omega"], + baseline = options.pop("baseline_voltage"), + duration = options.pop("pulse_duration"), + height = options.pop("pulse_height"), + phase = options.pop("pulse_phase"), + nmax = options["nmax"], + ) + options.pop("pulse_separation") + elif pulse_shape == "gauss_symmetric": + vdc, fourier_coef = fourier_coef_gauss_symmetric( + omega = options["omega"], + baseline = options.pop("baseline_voltage"), + duration = options.pop("pulse_duration"), + height = options.pop("pulse_height"), + phase = options.pop("pulse_phase"), + nmax = options["nmax"], + ) + options.pop("pulse_separation") + elif pulse_shape == "updown_gauss": + vdc, fourier_coef = fourier_coef_updown_gauss( + omega = options["omega"], + baseline = options.pop("baseline_voltage"), + duration = options.pop("pulse_duration"), + height = options.pop("pulse_height"), + phase = options.pop("pulse_phase"), + separation = options.pop("pulse_separation"), + nmax = options["nmax"], + ) + else: + raise ValueError(f"invalid pulse shape: {pulse_shape}") # Generate data dm = DataManager() -- GitLab