Skip to content

Commit ed2baa5

Browse files
committed
Separate dispatch for dropping trailing 1s (fixes for non-1 based arrays)
1 parent 379f18e commit ed2baa5

File tree

2 files changed

+10
-6
lines changed

2 files changed

+10
-6
lines changed

base/abstractarray.jl

+6-2
Original file line numberDiff line numberDiff line change
@@ -853,10 +853,14 @@ _to_subscript_indices{T}(A::AbstractArray{T,0}, i::Int) = () # TODO: REMOVE FOR
853853
_to_subscript_indices{T}(A::AbstractArray{T,0}, I::Int...) = () # TODO: DEPRECATE FOR #14770
854854
function _to_subscript_indices{T,N}(A::AbstractArray{T,N}, I::Int...) # TODO: DEPRECATE FOR #14770
855855
@_inline_meta
856-
J, _ = IteratorsMD.split(I, Val{N}) # (maybe) drop any trailing indices
857-
sz = _remaining_size(J, size(A)) # compute trailing size (overlapping the final index)
856+
J, Jrem = IteratorsMD.split(I, Val{N})
857+
_to_subscript_indices(A, J, Jrem)
858+
end
859+
function _to_subscript_indices(A::AbstractArray, J::Tuple, Jrem::Tuple{})
860+
sz = _remaining_size(J, indices(A)) # compute trailing size (overlapping the final index)
858861
(front(J)..., _unsafe_ind2sub(sz, last(J))...) # (maybe) extend the last index
859862
end
863+
_to_subscript_indices(A, J::Tuple, Jrem::Tuple) = J # already bounds-checked, safe to drop
860864
_to_subscript_indices{T,N}(A::AbstractArray{T,N}, I::Vararg{Int,N}) = I
861865
_remaining_size(::Tuple{Any}, t::Tuple) = t
862866
_remaining_size(h::Tuple, t::Tuple) = (@_inline_meta; _remaining_size(tail(h), tail(t)))

test/offsetarray.jl

+4-4
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,10 @@ S = OffsetArray(view(A0, 1:2, 1:2), (-1,2)) # LinearSlow
2222
@test_throws ErrorException size(A, 1)
2323

2424
# Scalar indexing
25-
@test A[0,3] == A[1] == S[0,3] == S[1] == 1
26-
@test A[1,3] == A[2] == S[1,3] == S[2] == 2
27-
@test A[0,4] == A[3] == S[0,4] == S[3] == 3
28-
@test A[1,4] == A[4] == S[1,4] == S[4] == 4
25+
@test A[0,3] == A[1] == A[0,3,1] == S[0,3] == S[1] == S[0,3,1] == 1
26+
@test A[1,3] == A[2] == A[1,3,1] == S[1,3] == S[2] == S[1,3,1] == 2
27+
@test A[0,4] == A[3] == A[0,4,1] == S[0,4] == S[3] == S[0,4,1] == 3
28+
@test A[1,4] == A[4] == A[1,4,1] == S[1,4] == S[4] == S[1,4,1] == 4
2929
@test_throws BoundsError A[1,1]
3030
@test_throws BoundsError S[1,1]
3131

0 commit comments

Comments
 (0)