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