11
2- import LazyStack
2+ using LazyStack
33
44# for stack_iter
55LazyStack. no_wraps (a:: KeyedArray ) = LazyStack. no_wraps (NamedDims. unname (parent (a)))
@@ -14,35 +14,35 @@ function LazyStack.rewrap_like(A, a::NamedTuple)
1414end
1515
1616# tuple of arrays
17- function LazyStack. stack (x:: Tuple{Vararg{<:KeyedArray}} )
17+ function LazyStack. lazystack (x:: Tuple{Vararg{<:KeyedArray}} )
1818 KeyedArray (LazyStack. stack (map (parent, x)), stack_keys (x))
1919end
2020
2121stack_keys (xs:: Tuple{Vararg{<:KeyedArray}} ) =
2222 (keys_or_axes (first (xs))... , Base. OneTo (length (xs)))
2323
2424# array of arrays: first strip off outer containers...
25- function LazyStack. stack (xs:: KeyedArray{<:AbstractArray} )
25+ function LazyStack. lazystack (xs:: KeyedArray{<:AbstractArray} )
2626 KeyedArray (LazyStack. stack (parent (xs)), stack_keys (xs))
2727end
28- function LazyStack. stack (xs:: KeyedArray{<:AbstractArray,N,<:NamedDimsArray{L}} ) where {L,N}
28+ function LazyStack. lazystack (xs:: KeyedArray{<:AbstractArray,N,<:NamedDimsArray{L}} ) where {L,N}
2929 data = LazyStack. stack (parent (parent (xs)))
3030 KeyedArray (LazyStack. ensure_named (data, LazyStack. getnames (xs)), stack_keys (xs))
3131end
32- function LazyStack. stack (xs:: NamedDimsArray{L,<:AbstractArray,N,<:KeyedArray} ) where {L,N}
32+ function LazyStack. lazystack (xs:: NamedDimsArray{L,<:AbstractArray,N,<:KeyedArray} ) where {L,N}
3333 data = LazyStack. stack (parent (parent (xs)))
3434 LazyStack. ensure_named (KeyedArray (data, stack_keys (xs)), LazyStack. getnames (xs))
3535end
3636
3737# ... then deal with inner ones:
38- function LazyStack. stack (x:: AT ) where {AT <: AbstractArray{<:KeyedArray{T,IN},ON} } where {T,IN,ON}
38+ function LazyStack. lazystack (x:: AT ) where {AT <: AbstractArray{<:KeyedArray{T,IN},ON} } where {T,IN,ON}
3939 KeyedArray (LazyStack. Stacked {T, IN+ON, AT} (x), stack_keys (x))
4040end
41- function LazyStack. stack (x:: AT ) where {AT <: AbstractArray{<:KeyedArray{T,IN,IT},ON} } where {IT<: NamedDimsArray } where {T,IN,ON}
41+ function LazyStack. lazystack (x:: AT ) where {AT <: AbstractArray{<:KeyedArray{T,IN,IT},ON} } where {IT<: NamedDimsArray } where {T,IN,ON}
4242 data = LazyStack. Stacked {T, IN+ON, AT} (x)
4343 KeyedArray (NamedDimsArray (data, LazyStack. getnames (x)), stack_keys (x))
4444end
45- function LazyStack. stack (x:: AT ) where {AT <: AbstractArray{<:NamedDimsArray{L,T,IN,IT},ON} } where {IT<: KeyedArray } where {T,IN,ON,L}
45+ function LazyStack. lazystack (x:: AT ) where {AT <: AbstractArray{<:NamedDimsArray{L,T,IN,IT},ON} } where {IT<: KeyedArray } where {T,IN,ON,L}
4646 data = LazyStack. Stacked {T, IN+ON, AT} (x)
4747 NamedDimsArray (KeyedArray (data, stack_keys (x)), LazyStack. getnames (x))
4848end
0 commit comments