Added Wiener deconvolution.

pes_to_spec/test/offline_analysis.py

@@ -18,7 +18,7 @@ matplotlib.use('Agg')
 
 import matplotlib.pyplot as plt
 from matplotlib.gridspec import GridSpec
-from mpl_toolkits.axes_grid.inset_locator import InsetPosition
+from mpl_toolkits.axes_grid1.inset_locator import InsetPosition
 import seaborn as sns
 import scipy
 from scipy.signal import fftconvolve
@@ -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)