From 0cdb914b5764f692189ed2bc395e3b09ead758e4 Mon Sep 17 00:00:00 2001 From: Ian Jauslin Date: Fri, 27 May 2022 16:09:17 -0400 Subject: savefile and initfile --- src/navier-stokes.c | 19 ++++++++++++++++--- 1 file changed, 16 insertions(+), 3 deletions(-) (limited to 'src/navier-stokes.c') diff --git a/src/navier-stokes.c b/src/navier-stokes.c index 7c4ecb3..12cba85 100644 --- a/src/navier-stokes.c +++ b/src/navier-stokes.c @@ -1,4 +1,5 @@ #include "navier-stokes.h" +#include "io.h" #include #include @@ -15,7 +16,8 @@ int uk( _Complex double* u0, _Complex double* g, unsigned int print_freq, - unsigned int nthreads + unsigned int nthreads, + FILE* savefile ){ _Complex double* u; _Complex double* tmp1; @@ -65,6 +67,9 @@ int uk( } } + // save final entry to savefile + write_u(u, K1, K2, savefile); + ns_free_tmps(u, tmp1, tmp2, tmp3, fft1, fft2, ifft); return(0); } @@ -82,7 +87,8 @@ int eea( _Complex double* u0, _Complex double* g, unsigned int print_freq, - unsigned int nthreads + unsigned int nthreads, + FILE* savefile ){ _Complex double* u; _Complex double* tmp1; @@ -126,6 +132,9 @@ int eea( } } + // save final entry to savefile + write_u(u, K1, K2, savefile); + ns_free_tmps(u, tmp1, tmp2, tmp3, fft1, fft2, ifft); return(0); } @@ -142,7 +151,8 @@ int quiet( double L, _Complex double* u0, _Complex double* g, - unsigned int nthreads + unsigned int nthreads, + FILE* savefile ){ _Complex double* u; _Complex double* tmp1; @@ -162,6 +172,9 @@ int quiet( ins_step(u, K1, K2, N1, N2, nu, delta, L, g, fft1, fft2, ifft, tmp1, tmp2, tmp3); } + // save final entry to savefile + write_u(u, K1, K2, savefile); + ns_free_tmps(u, tmp1, tmp2, tmp3, fft1, fft2, ifft); return(0); } -- cgit v1.2.3-54-g00ecf