opxrd/plots: Added fourier transform suptitles

aimat-lab · Dec 22, 2024 · a2bd92f · a2bd92f
1 parent 783f7cc
commit a2bd92f
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 30 deletions.
diff --git a/opxrd/analysis/plotter.py b/opxrd/analysis/plotter.py
@@ -94,7 +94,6 @@ def _fourier_plots(self, x, y_list: list[NDArray], msg: str, figname: str, y_nam
         if len(y_list) == 0:
             raise ValueError('No y data provided for Fourier Transform plot')
 
-        print_text(msg)
         xf, _ = compute_standardized_fourier(x, y_list[0])
         yf_list = []
         for y in y_list:
@@ -118,6 +117,8 @@ def _fourier_plots(self, x, y_list: list[NDArray], msg: str, figname: str, y_nam
         ax2.set_yscale(f'log')
         ax2.set_title('Fourier Transform')
 
+        fig.suptitle(msg)
+
         if y_names:
             ax1.legend(y_names, ncol=2, loc='upper right',fontsize=8)
             ax2.legend(y_names, ncol=2, loc='upper right',fontsize=8)