allow to specify 3 vectors as direction

osyris-project · Apr 5, 2024 · 903297a · 903297a
1 parent d335bef
commit 903297a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/osyris/plot/direction.py b/src/osyris/plot/direction.py
@@ -126,6 +126,11 @@ def get_direction(direction=None, dataset=None, dx=None, dy=None, origin=None):
         v = direction.cross(dir_vecs["pos_u"])
         v.name = "pos_v"
         dir_vecs["pos_v"] = v
+    elif isinstance(direction, dict):
+        dir_vecs = direction.copy()
+        assert "normal" in dir_vecs
+        assert "pos_u" in dir_vecs
+        assert "pos_v" in dir_vecs
     else:
         raise ValueError(f"Bad direction for slice: {direction}.")