File tree Expand file tree Collapse file tree 1 file changed +1
-9
lines changed Expand file tree Collapse file tree 1 file changed +1
-9
lines changed Original file line number Diff line number Diff line change @@ -2986,14 +2986,6 @@ def calc_arrows(UVW):
2986
2986
# Normalize rows of UVW
2987
2987
norm = np .linalg .norm (UVW , axis = 1 )
2988
2988
2989
- # If any row of UVW is all zeros, don't make a quiver for it
2990
- mask = norm > 0
2991
- XYZ = XYZ [mask ]
2992
- if normalize :
2993
- UVW = UVW [mask ] / norm [mask ].reshape ((- 1 , 1 ))
2994
- else :
2995
- UVW = UVW [mask ]
2996
-
2997
2989
if len (XYZ ) > 0 :
2998
2990
# compute the shaft lines all at once with an outer product
2999
2991
shafts = (XYZ - np .multiply .outer (shaft_dt , UVW )).swapaxes (0 , 1 )
@@ -3006,7 +2998,7 @@ def calc_arrows(UVW):
3006
2998
# transpose to get a list of lines
3007
2999
heads = heads .swapaxes (0 , 1 )
3008
3000
3009
- lines = [* shafts , * heads ]
3001
+ lines = [* shafts , * heads [:: 2 ], * heads [ 1 :: 2 ] ]
3010
3002
else :
3011
3003
lines = []
3012
3004
You can’t perform that action at this time.
0 commit comments