add slicing of tuples of matrices for plotattributes (#4109)

* add slicing of tuples of matrices

* remove dot

* format files
This commit is contained in:
Simon Christ 2022-02-18 21:12:44 +01:00 committed by GitHub
parent ed9f0bcab3
commit 51358717ee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 5 deletions

View File

@ -1748,15 +1748,17 @@ function slice_arg!(
remove_pair::Bool,
)
v = get(plotattributes_in, k, plotattributes_out[k])
plotattributes_out[k] =
if haskey(plotattributes_in, k) &&
typeof(v) <: AMat &&
!isempty(v) &&
!(k in _plot_args)
plotattributes_out[k] = if haskey(plotattributes_in, k) && !(k in _plot_args)
if typeof(v) <: AMat && !isempty(v)
slice_arg(v, idx)
elseif typeof(v) <: NTuple{2,AMat}
(slice_arg(v[1], idx), slice_arg(v[2], idx))
else
v
end
else
v
end
if remove_pair
RecipesPipeline.reset_kw!(plotattributes_in, k)
end

View File

@ -32,5 +32,7 @@ end
@test plot(data4; NamedTuple{tuple(attribute)}(Ref(mat))...)[1][i][attribute] ==
[2(i - 1) + 1, 2i]
end
@test plot(data4, ribbon = (mat, mat))[1][i][:ribbon] ==
([2(i - 1) + 1, 2i], [2(i - 1) + 1, 2i])
end
end