Added Wiener deconvolution.

pes_to_spec/test/offline_analysis.py

@@ -250,7 +250,7 @@ def main():
     fig = plt.figure(figsize=(12, 8))
     gs = GridSpec(1, 1)
     ax = fig.add_subplot(gs[0, 0])
-    plt.scatter(np.fft.fftshift(model.wiener_energy), np.fft.fftshift(np.absolute(model.wiener_filter_ft)))
+    plt.plot(np.fft.fftshift(model.wiener_energy_ft), np.fft.fftshift(np.absolute(model.wiener_filter_ft)))
     ax.set(title=f"",
            xlabel=r"Reciprocal energy [1/eV]",
            ylabel="Filter intensity [a.u.]",
@@ -262,13 +262,10 @@ def main():
     fig = plt.figure(figsize=(12, 8))
     gs = GridSpec(1, 1)
     ax = fig.add_subplot(gs[0, 0])
-    de = spec_raw_pe[0, -1] - spec_raw_pe[0,0]
-    de = np.linspace(-0.5*de, 0.5*de, spec_raw_pe.shape[1])
-    plt.scatter(de, np.fft.fftshift(np.absolute(model.wiener_filter)))
+    plt.plot(model.wiener_energy, np.fft.fftshift(np.absolute(model.wiener_filter)))
     ax.set(title=f"",
            xlabel=r"Energy [eV]",
            ylabel="Filter value [a.u.]",
-           yscale='log',
            )
     fig.savefig(os.path.join(args.directory, "wiener.png"))
     plt.close(fig)