From 777a51be49e7c29c45963e9e16f07e00a69a880a Mon Sep 17 00:00:00 2001
From: Ian Jauslin <ian@jauslin.org>
Date: Thu, 26 May 2022 21:11:56 -0400
Subject: Average energy

---
 src/navier-stokes.c | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'src/navier-stokes.c')

diff --git a/src/navier-stokes.c b/src/navier-stokes.c
index 24b4508..531398d 100644
--- a/src/navier-stokes.c
+++ b/src/navier-stokes.c
@@ -99,6 +99,9 @@ int energy(
   // copy initial condition
   copy_u(u, u0, K1, K2);
 
+  // init running average
+  avg=0;
+
   // iterate
   for(t=0;t<nsteps;t++){
     ins_step(u, K1, K2, N1, N2, nu, delta, L, g, fft1, fft2, ifft, tmp1, tmp2, tmp3);
@@ -111,8 +114,13 @@ int energy(
 	}
       }
 
-      fprintf(stderr,"%d % .8e % .8e\n",t,t*delta, energy);
-      printf("%8d % .15e % .15e\n",t,t*delta,energy);
+      // running average
+      if(t>0){
+	avg=avg-(avg-energy)/t;
+      }
+
+      fprintf(stderr,"%d % .8e % .8e % .8e\n",t,t*delta, energy, avg);
+      printf("%8d % .15e % .15e % .15e\n",t,t*delta,energy,avg);
     }
   }
 
-- 
cgit v1.2.3-54-g00ecf