From 3d7fb6412e90560f4fb06b22673d98e8c14ec261 Mon Sep 17 00:00:00 2001 From: Craig Warren Date: Tue, 24 Nov 2020 17:49:51 +0000 Subject: [PATCH] Added method to allow snapshot as vti or h5 --- gprMax/model_build_run.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/gprMax/model_build_run.py b/gprMax/model_build_run.py index 1587f6ce..3504b228 100644 --- a/gprMax/model_build_run.py +++ b/gprMax/model_build_run.py @@ -232,10 +232,12 @@ class ModelBuildRun: for i, snap in enumerate(self.G.snapshots): fn = snapshotdir / Path(snap.filename) - snap.filename = fn.with_suffix('.vti') + snap.filename = fn.with_suffix(snap.fileext) pbar = tqdm(total=snap.vtkdatawritesize, leave=True, unit='byte', - unit_scale=True, desc=f'Writing snapshot file {i + 1} of {len(self.G.snapshots)}, {snap.filename.name}', ncols=get_terminal_width() - 1, file=sys.stdout, disable=not config.sim_config.general['progressbars']) - snap.write_vtk_imagedata(pbar, self.G) + unit_scale=True, desc=f'Writing snapshot file {i + 1} of {len(self.G.snapshots)}, {snap.filename.name}', + ncols=get_terminal_width() - 1, file=sys.stdout, + disable=not config.sim_config.general['progressbars']) + snap.write_file(pbar, self.G) pbar.close() logger.info('')