From 16fbe0fb8b0a6d7ca67fe93ec77368a55f52fe2d Mon Sep 17 00:00:00 2001
From: Holger Vogt <holger.vogt@uni-due.de>
Date: Mon, 10 Mar 2025 12:53:40 +0100
Subject: [PATCH] Create a vector thdnm from THD calculation, e.g. thd11 New
 vector fournosave: if set, suppress creation of vectors thd11 and fourier,
 usefull when writing a raw file.

---
 src/frontend/fourier.c | 65 ++++++++++++++++++++++++++----------------
 1 file changed, 41 insertions(+), 24 deletions(-)

diff --git a/src/frontend/fourier.c b/src/frontend/fourier.c
index 47b7e6158..d66efe3ee 100644
--- a/src/frontend/fourier.c
+++ b/src/frontend/fourier.c
@@ -50,6 +50,7 @@ fourier(wordlist *wl, struct plot *current_plot)
     char xbuf[20];
     int shift;
     int rv = 1;
+    bool foursave = TRUE;
 
     struct dvec *n;
     int newveccount = 1;
@@ -71,6 +72,8 @@ fourier(wordlist *wl, struct plot *current_plot)
         polydegree = 1;
     if (!cp_getvar("fourgridsize", CP_NUM, &fourgridsize, 0) || fourgridsize < 1)
         fourgridsize = DEF_FOURGRIDSIZE;
+    if (cp_getvar("fournosave", CP_BOOL, NULL, 0))
+        foursave = FALSE;
 
     time = current_plot->pl_scale;
     if (!isreal(time)) {
@@ -188,34 +191,48 @@ fourier(wordlist *wl, struct plot *current_plot)
             }
             fputs("\n", cp_out);
 
-            /* create and assign a new vector n */
-            /* with size 3 * nfreqs in current plot */
-            /* generate name for new vector, using vec->name */
-            n = dvec_alloc(tprintf("fourier%d%d", callstof, newveccount),
-                           SV_NOTYPE,
-                           VF_REAL | VF_PERMANENT,
-                           3 * nfreqs, NULL);
+            if (foursave) {
+                /* create a vector for THD */
+                n = dvec_alloc(tprintf("thd%d%d", callstof, newveccount),
+                    SV_NOTYPE,
+                    VF_REAL | VF_PERMANENT,
+                    1, NULL);
 
-            n->v_numdims = 2;
-            n->v_dims[0] = 3;
-            n->v_dims[1] = nfreqs;
+                n->v_numdims = 1;
 
-            vec_new(n);
+                vec_new(n);
 
-            /* store data in vector: freq, mag, phase */
-            for (i = 0; i < nfreqs; i++) {
-                n->v_realdata[i] = freq[i];
-                n->v_realdata[i + nfreqs] = mag[i];
-                n->v_realdata[i + 2 * nfreqs] = phase[i];
+                n->v_realdata[0] = thd;
+
+                /* create and assign a new vector n */
+                /* with size 3 * nfreqs in current plot */
+                /* generate name for new vector, using vec->name */
+                n = dvec_alloc(tprintf("fourier%d%d", callstof, newveccount),
+                    SV_NOTYPE,
+                    VF_REAL | VF_PERMANENT,
+                    3 * nfreqs, NULL);
+
+                n->v_numdims = 2;
+                n->v_dims[0] = 3;
+                n->v_dims[1] = nfreqs;
+
+                vec_new(n);
+
+                /* store data in vector: freq, mag, phase */
+                for (i = 0; i < nfreqs; i++) {
+                    n->v_realdata[i] = freq[i];
+                    n->v_realdata[i + nfreqs] = mag[i];
+                    n->v_realdata[i + 2 * nfreqs] = phase[i];
+                }
+                newveccount++;
+
+                if (polydegree) {
+                    tfree(timescale);
+                    tfree(data);
+                }
+                timescale = NULL;
+                data = NULL;
             }
-            newveccount++;
-
-            if (polydegree) {
-                tfree(timescale);
-                tfree(data);
-            }
-            timescale = NULL;
-            data = NULL;
         }
     }