Skip to content

Commit 903297a

Browse files
committed
allow to specify 3 vectors as direction
1 parent d335bef commit 903297a

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/osyris/plot/direction.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,11 @@ def get_direction(direction=None, dataset=None, dx=None, dy=None, origin=None):
126126
v = direction.cross(dir_vecs["pos_u"])
127127
v.name = "pos_v"
128128
dir_vecs["pos_v"] = v
129+
elif isinstance(direction, dict):
130+
dir_vecs = direction.copy()
131+
assert "normal" in dir_vecs
132+
assert "pos_u" in dir_vecs
133+
assert "pos_v" in dir_vecs
129134
else:
130135
raise ValueError(f"Bad direction for slice: {direction}.")
131136

0 commit comments

Comments
 (0)