From e9cad56d5d4f9a26ada81f197dfbff26f9c04047 Mon Sep 17 00:00:00 2001
From: erikkaashoek <erik@kaashoek.com>
Date: Fri, 5 Mar 2021 11:26:28 +0100
Subject: [PATCH] Waterfall from temp trace when in calculation mode

---
 plot.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/plot.c b/plot.c
index 8b6e12e..2ac592f 100644
--- a/plot.c
+++ b/plot.c
@@ -2321,7 +2321,11 @@ static void update_waterfall(void){
     ili9341_read_memory(OFFSETX, i  , w_width, 1, w_width*1, spi_buffer);
            ili9341_bulk(OFFSETX, i+1, w_width, 1);
   }
-  index_t *index = trace_index[TRACE_ACTUAL];
+  index_t *index;
+  if (setting.average == AV_OFF)
+    index = trace_index[TRACE_ACTUAL];
+  else
+    index = trace_index[TRACE_TEMP];
   int j = 0;
   for (i=0; i< sweep_points; i++) {			// Add new topline
     uint16_t color;