bugfixes vtk

pyrocko · Oct 27, 2023 · 3f06448 · 3f06448
1 parent 8e419e2
commit 3f06448
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/lassie/tracers/fast_marching/fast_marching.py b/lassie/tracers/fast_marching/fast_marching.py
@@ -260,8 +260,8 @@ def export_vtk(self, filename: Path, reference: Location | None = None) -> None:
             str(filename),
             self._east_coords + offset[0],
             self._north_coords + offset[1],
-            -(self._depth_coords + offset[2]),
-            pointData={"travel_times": self.travel_times},
+            np.array((-self._depth_coords + offset[2])[::-1]),
+            pointData={"travel_time": np.array(self.travel_times[:, :, ::-1])},
         )
         logger.debug(
             "vtk: exported travel times of %s to %s",

diff --git a/lassie/tracers/fast_marching/velocity_models.py b/lassie/tracers/fast_marching/velocity_models.py
@@ -247,8 +247,8 @@ def export_vtk(self, filename: Path, reference: Location | None = None) -> None:
             str(filename),
             self._east_coords + offset[0],
             self._north_coords + offset[1],
-            -(self._depth_coords + offset[2]),
-            pointData={"velocity": self._velocity_model},
+            np.array((-self._depth_coords + offset[2])[::-1]),
+            pointData={"velocity": np.array(self._velocity_model[:, :, ::-1])},
         )
         logger.info("vtk: exported velocity model to %s", out_file)