add metadata to plotspectrumdata output

src/est/core/process/plotspectrumdata.py

@@ -11,6 +11,19 @@ _Y_TO_X_NAME = {
     "ft_mag": "R",
     "noise_savgol": "energy",
 }
+_Y_TO_INFO_ATTRS = {
+    "normalized_mu": ["e0", "edge_step"],
+    "flatten_mu": ["e0", "edge_step"],
+    "noise_savgol": ["raw_noise_savgol", "norm_noise_savgol", "edge_step", "e0"],
+}
+_Y_TO_HLINE_ATTRS = {"noise_savgol": ["raw_noise_savgol"]}
+_Y_TO_VLINE_ATTRS = {"noise_savgol": ["noise_e_min", "noise_e_max"]}
+_LARCH_ATTRS = ("noise_e_min", "noise_e_max")
+_INFO_ATTRS_TO_NAMES = {
+    "raw_noise_savgol": "raw_noise",
+    "norm_noise_savgol": "norm_noise",
+    "edge_step": "edge_step",
+}
 _LABELS = {
     "energy": "Energy (eV)",
     "mu": "μ(E)",
@@ -30,40 +43,60 @@ def process_plotspectrumdata(
 ) -> Dict[str, dict]:
     kweight = spectrum.larch_dict["xftf_k_weight"]
     params = {"kweight": kweight, "kweightp1": kweight + 1}
-
     plot_data = dict()
     if not plot_names:
         plot_names = ("flatten_mu", "chi_weighted_k", "ft_mag", "noise_savgol")
-
     for yname in plot_names:
-        ylabel = _LABELS[yname]
-        xname = _Y_TO_X_NAME[yname]
-        xlabel = _LABELS[xname]
+        plot_data[yname] = get_plotspectrumdata(spectrum, yname, params)
+    return plot_data
 
-        if yname == "ft_mag":
-            x = spectrum.ft["FTRadius"]
-            y = spectrum.ft["FTIntensity"]
-            xlabel = _LABELS["R"]
-            ylabel = _LABELS["ft_mag"]
-        else:
-            try:
-                x = getattr(spectrum, xname)
-                y = getattr(spectrum, yname)
-            except Exception:
-                breakpoint()
 
-        xlabel = xlabel.format(**params)
-        ylabel = ylabel.format(**params)
+def get_plotspectrumdata(spectrum, yname, params):
+    xname = _Y_TO_X_NAME[yname]
+    if yname == "ft_mag":
+        x = spectrum.ft["FTRadius"]
+        y = spectrum.ft["FTIntensity"]
+    else:
+        x = getattr(spectrum, xname)
+        y = getattr(spectrum, yname)
 
-        plot_data[yname] = {
-            "name": yname,
-            "x": x,
-            "y": y,
-            "xlabel": xlabel,
-            "ylabel": ylabel,
-            "ylog": yname == "noise_savgol",
-        }
-    return plot_data
+    xlabel = _LABELS[xname]
+    ylabel = _LABELS[yname]
+    xlabel = xlabel.format(**params)
+    ylabel = ylabel.format(**params)
+
+    info = dict()
+    for attr in _Y_TO_INFO_ATTRS.get(yname, list()):
+        value = get_spectrum_value(spectrum, attr)
+        key = _INFO_ATTRS_TO_NAMES.get(attr, attr)
+        info[key] = value
+
+    hlines = [
+        get_spectrum_value(spectrum, attr)
+        for attr in _Y_TO_HLINE_ATTRS.get(yname, list())
+    ]
+    vlines = [
+        get_spectrum_value(spectrum, attr)
+        for attr in _Y_TO_VLINE_ATTRS.get(yname, list())
+    ]
+
+    return {
+        "name": yname,
+        "x": x,
+        "y": y,
+        "xlabel": xlabel,
+        "ylabel": ylabel,
+        "info": info,
+        "hlines": hlines,
+        "vlines": vlines,
+    }
+
+
+def get_spectrum_value(spectrum, attr):
+    if attr in _LARCH_ATTRS:
+        return spectrum.larch_dict[attr]
+    else:
+        return getattr(spectrum, attr)
 
 
 class PlotSpectrumData(

src/est/tests/test_plotspectrumdata.py

@@ -89,8 +89,6 @@ def test_larch_plotspectrumdata(filename_cu_from_pymca):
             ax.plot(data["x"], data["y"])
             ax.set_xlabel(data["xlabel"])
             ax.set_ylabel(data["ylabel"])
-            if data["ylog"]:
-                ax.set_yscale("log")
         plt.show()