From adfb7c5ec1a3a34b1e6e4f7c1e1a0ac62101df3a Mon Sep 17 00:00:00 2001 From: David Minton Date: Wed, 28 Feb 2024 14:19:19 -0500 Subject: [PATCH] Make sure to save the first frame on a non-restart run. --- swiftest/simulation_class.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/swiftest/simulation_class.py b/swiftest/simulation_class.py index abc552072..a7c2d0438 100644 --- a/swiftest/simulation_class.py +++ b/swiftest/simulation_class.py @@ -448,9 +448,11 @@ def run(self, return # Save initial conditions - if not self.restart: + if self.restart: + self.save(framenum=-1) + else: self.clean() - self.save() + self.save(framenum=0) # Write out the current parameter set before executing run self.write_param(verbose=False,**kwargs) @@ -2952,7 +2954,7 @@ def read_output_file(self, param_tmp['BIN_OUT'] = self.simdir / self.param['NC_IN'] self.init_cond = io.swiftest2xr(param_tmp, verbose=False, dask=dask) else: - self.init_cond = self.data.isel(time=0) + self.init_cond = self.data.isel(time=[0]).copy(deep=True) if self.read_encounters: self.read_encounter_file(dask=dask) @@ -3124,7 +3126,7 @@ def save(self, if not self.simdir.exists(): self.simdir.mkdir(parents=True, exist_ok=True) - self.init_cond = self.data.copy(deep=True) + self.init_cond = self.data.isel(time=[framenum]).copy(deep=True) if codename == "Swiftest": infile_name = Path(self.simdir) / param['NC_IN']