Skip to content

Commit 8e3284f

Browse files
committed
Slice axes
1 parent 7e0e5f0 commit 8e3284f

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/cartesianproduct.jl

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,6 +143,11 @@ function Base.checkindex(::Type{Bool}, inds::CartesianProductUnitRange, i::Carte
143143
return checkindex(Bool, arg1(inds), arg1(i)) && checkindex(Bool, arg2(inds), arg2(i))
144144
end
145145

146+
const CartesianProductOneTo{T,P<:CartesianProduct,R<:Base.OneTo{T}} = CartesianProductUnitRange{T,P,R}
147+
Base.axes(S::Base.Slice{<:CartesianProductOneTo}) = (S.indices,)
148+
Base.axes1(S::Base.Slice{<:CartesianProductOneTo}) = S.indices
149+
Base.unsafe_indices(S::Base.Slice{<:CartesianProductOneTo}) = (S.indices,)
150+
146151
function Base.getindex(a::CartesianProductUnitRange, I::CartesianProduct)
147152
prod = cartesianproduct(a)
148153
prod_I = arg1(prod)[arg1(I)] × arg2(prod)[arg2(I)]

0 commit comments

Comments
 (0)