From 3dcd44c3eefc7d2ed911622f31a235e86e3a0db0 Mon Sep 17 00:00:00 2001
From: Dominik Demuth <dominik.demuth@physik.tu-darmstadt.de>
Date: Fri, 19 May 2023 11:37:24 +0200
Subject: [PATCH] restore optionality to figure saving

---
 src/nmreval/io/fcbatchreader.py | 53 +++++++++++++++++----------------
 1 file changed, 27 insertions(+), 26 deletions(-)

diff --git a/src/nmreval/io/fcbatchreader.py b/src/nmreval/io/fcbatchreader.py
index 9bb09fb..f67a661 100644
--- a/src/nmreval/io/fcbatchreader.py
+++ b/src/nmreval/io/fcbatchreader.py
@@ -213,7 +213,7 @@ class FCReader:
                     except KeyError:
                         self.f_params[k] = [new_entry]
 
-                if True:  # save_fits or save_fig:
+                if save_fits or save_fig:
                     xplot = np.geomspace(v.x[0], v.x[-1], num=10*len(v.x))
                     yplot = FCReader.kww(xplot, *p0)
                     save_name = f'{filename.stem}_{k:011.2f}'.replace('.', 'p') + '.dat'
@@ -222,33 +222,34 @@ class FCReader:
                         np.savetxt(fit_path.joinpath(save_name), np.c_[xplot, yplot],
                                    header=header+'\t'.join([f'{p}+/-{err}' for p, err in zip(p0, perr)]))
 
-                    img_file = image_path.joinpath(save_name).with_suffix(".png")
+                    if save_fig:
+                        img_file = image_path.joinpath(save_name).with_suffix(".png")
 
-                    gnuplot_args = [
-                        'set terminal png;',
-                        f'set output "{img_file}";',
-                        f'set title "f = {k:.4g} Hz\\n{p0[2]:.4g}(+/-{perr[2]:.4g}) beta: {p0[3]:.4g}(+/-{perr[3]:.4g})";',
-                        'set xlabel "t / s";',
-                        'set logscale x;',
-                        'set format x "10^{{%L}}";',
-                        'set ylabel "M";',
-                        'set key off;',
-                        f'plot "{data_path.joinpath(save_name)}" with points pointtype 5, "{fit_path.joinpath(save_name)}" with lines;',
-                    ]
+                        gnuplot_args = [
+                            'set terminal png;',
+                            f'set output "{img_file}";',
+                            f'set title "f = {k:.4g} Hz\\n{p0[2]:.4g}(+/-{perr[2]:.4g}) beta: {p0[3]:.4g}(+/-{perr[3]:.4g})";',
+                            'set xlabel "t / s";',
+                            'set logscale x;',
+                            'set format x "10^{{%L}}";',
+                            'set ylabel "M";',
+                            'set key off;',
+                            f'plot "{data_path.joinpath(save_name)}" with points pointtype 5, "{fit_path.joinpath(save_name)}" with lines;',
+                        ]
 
-                    try:
-                        proc = subprocess.Popen(
-                            ['gnuplot', '-p'],
-                            shell=True,
-                            stdin=subprocess.PIPE,
-                            encoding='utf8',
-                        )
-                        for args in gnuplot_args:
-                            proc.stdin.write(args)
-                        proc.stdin.write('quit\n')
-                        proc.stdin.flush()
-                    except Exception as e:
-                        logger.error(f'saving image {save_name} failed', e)
+                        try:
+                            proc = subprocess.Popen(
+                                ['gnuplot', '-p'],
+                                shell=True,
+                                stdin=subprocess.PIPE,
+                                encoding='utf8',
+                            )
+                            for args in gnuplot_args:
+                                proc.stdin.write(args)
+                            proc.stdin.write('quit\n')
+                            proc.stdin.flush()
+                        except Exception as e:
+                            logger.error(f'saving image {save_name} failed', e)
 
             freqs = np.asanyarray(freqs)
             params = np.asanyarray(params)