From d91cc08685f1ed68bc9100e44c987cd1b43f0d4a Mon Sep 17 00:00:00 2001
From: Craig Warren <craig.warren@northumbria.ac.uk>
Date: Fri, 17 Feb 2023 11:35:09 +0000
Subject: [PATCH] Testing subgrid snapshots

---
 gprMax/model_build_run.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/gprMax/model_build_run.py b/gprMax/model_build_run.py
index 1b3862ad..fd066916 100644
--- a/gprMax/model_build_run.py
+++ b/gprMax/model_build_run.py
@@ -251,8 +251,9 @@ class ModelBuildRun:
 
         write_hdf5_outputfile(config.get_model_config().output_file_path_ext, self.G)
 
-        if self.G.snapshots:
-            save_snapshots(self.G)
+        for grid in [self.G] + self.G.subgrids:
+            if grid.snapshots:
+                save_snapshots(grid)
 
     def solve(self, solver):
         """Solve using FDTD method.