Resolve ambiguity in infinite banded subarray broadcaststyle (#180)

dlfivefifty · web-flow · commit 344da69a1caf · 2024-06-04T10:52:29.000+08:00
* Resolve ambiguity in infinite banded subarray broadcaststyle

* Update test_infbanded.jl
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "InfiniteLinearAlgebra"
 uuid = "cde9dba0-b1de-11e9-2c62-0bab9446c55c"
-version = "0.8"
+version = "0.8.1"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"
diff --git a/src/InfiniteLinearAlgebra.jl b/src/InfiniteLinearAlgebra.jl
@@ -19,7 +19,7 @@ import ArrayLayouts: AbstractBandedLayout, AbstractQLayout, AdjQRPackedQLayout,
                      rowsupport, sub_materialize, subdiagonaldata, sublayout, supdiagonaldata, transposelayout,
                      triangulardata, triangularlayout, zero!, materialize!
 
-import BandedMatrices: BandedColumns, BandedMatrix, BandedMatrix, _BandedMatrix,
+import BandedMatrices: BandedColumns, BandedMatrix, BandedMatrix, _BandedMatrix, AbstractBandedMatrix,
                        _BandedMatrix, _BandedMatrix, _banded_qr, _banded_qr!, _default_banded_broadcast, banded_chol!,
                        banded_similar, bandedcolumns, bandeddata, bandwidths, bandwidths
 
@@ -32,7 +32,7 @@ import BlockBandedMatrices: AbstractBlockBandedLayout, BlockBandedMatrix, BlockS
 
 import FillArrays: AbstractFill, AbstractFillMatrix, axes_print_matrix_row, getindex_value
 
-import InfiniteArrays: AbstractInfUnitRange, InfAxes, InfRanges, InfStepRange, InfUnitRange, OneToInf, PosInfinity
+import InfiniteArrays: AbstractInfUnitRange, InfAxes, InfRanges, InfStepRange, InfUnitRange, OneToInf, PosInfinity, InfIndexRanges
 
 import Infinities: InfiniteCardinal, Infinity
 
diff --git a/src/banded/infbanded.jl b/src/banded/infbanded.jl
@@ -1,3 +1,5 @@
+BroadcastStyle(::Type{<:SubArray{<:Any,2,<:AbstractBandedMatrix,<:Tuple{<:InfIndexRanges,<:InfIndexRanges}}})= LazyArrayStyle{2}()
+
 _BandedMatrix(data::AbstractMatrix{T}, ::Infinity, l, u) where T = _BandedMatrix(data, ℵ₀, l, u)
 
 ###
diff --git a/test/test_infbanded.jl b/test/test_infbanded.jl
@@ -64,6 +64,7 @@ using Base: oneto
         @test T[2:∞,3:∞] isa SubArray
         @test exp.(T) isa BroadcastMatrix
         @test exp.(T)[2:∞,3:∞] isa SubArray
+        @test exp.(T[2:∞,3:∞]) isa BroadcastMatrix
 
         B = LazyBandedMatrices.Bidiagonal(Fill(1,∞), Zeros(∞), :U)
         @test B[2:∞,3:∞] isa SubArray

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,5 @@`
	`1`	`+BroadcastStyle(::Type{<:SubArray{<:Any,2,<:AbstractBandedMatrix,<:Tuple{<:InfIndexRanges,<:InfIndexRanges}}})= LazyArrayStyle{2}()`
	`2`	`+`
`1`	`3`	`_BandedMatrix(data::AbstractMatrix{T}, ::Infinity, l, u) where T = _BandedMatrix(data, ℵ₀, l, u)`
`2`	`4`
`3`	`5`	`###`