Merge pull request #384 from volodia99/return-plot-1d

BUG: fix reading coordinates in `Fargo3DReader.read`
volodia99 · Dec 16, 2024 · c5ef2e7 · c5ef2e7
2 parents 026e270 + 681dc32
commit c5ef2e7
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/nonos/_readers/binary.py b/nonos/_readers/binary.py
@@ -600,14 +600,22 @@ def read(
         n1 = len(V["x1"]) - 1
         n2 = len(V["x2"]) - 1
         n3 = len(V["x3"]) - 1
-        grid_shape = n3, n1, n2
-        shift = n2 // 2
+        if geometry_str == "cylindrical":
+            grid_shape = n3, n1, n2
+            tuple_transposition = (1, 2, 0)
+            shift = n2 // 2
+        elif geometry_str == "spherical":
+            grid_shape = n2, n1, n3
+            tuple_transposition = (1, 0, 2)
+            shift = n3 // 2
+        else:
+            raise NotImplementedError(f"Geometry {geometry_str!r} is not supported")
 
         def _read_array(file: Path):
             return np.roll(
                 np.fromfile(file, dtype="float64")
                 .reshape(grid_shape)
-                .transpose(1, 2, 0),
+                .transpose(tuple_transposition),
                 shift,
                 axis=1,
             )