Long Data Profile: Optional Phase (#212)

Author: ax3l

lasy/profiles/longitudinal/longitudinal_profile_from_data.py

@@ -63,7 +63,10 @@ def __init__(self, data, lo, hi):
             # First find central frequency
             wavelength = data["axis"]
             spectral_intensity = data["intensity"]
-            spectral_phase = data["phase"]
+            if data.get("phase") is None:
+                spectral_phase = np.zeros_like(wavelength)
+            else:
+                spectral_phase = data["phase"]
             dt = data["dt"]
             cwl = np.sum(spectral_intensity * wavelength) / np.sum(spectral_intensity)
             cfreq = c / cwl
@@ -106,7 +109,10 @@ def __init__(self, data, lo, hi):
         elif data["datatype"] == "temporal":
             time = data["axis"]
             temporal_intensity = data["intensity"]
-            temporal_phase = data["phase"]
+            if data.get("phase") is None:
+                temporal_phase = np.zeros_like(time)
+            else:
+                temporal_phase = data["phase"]
             cwl = data["wavelength"]
 
         else: