BlockDiagonal

Author: dlfivefifty

examples/toeplitz.jl

@@ -109,4 +109,4 @@ A = BandedMatrix(-2 => Fill(1,∞), -1 => Fill(-im,∞), 1 => Fill(im,∞), 2 =>
 clf(); qlplot(A; x=range(-3,3; length=100), y=range(-3,3;length=100)); symbolplot(A; color=:black); title("Butterfly")
 clf(); qlplot(A; branch=branch(2), x=range(-3,3; length=100), y=range(-3,3;length=100)); symbolplot(A; color=:black); title("Butterfly, branch 2")
 clf(); qlplot(transpose(A); x=range(-3,3; length=100), y=range(-3,3;length=100)); symbolplot(A; color=:black); title("Butterfly")
-clf(); qlplot(transpose(A); branch=branch(2), x=range(-3,3; length=100), y=range(-3,3;length=100)); symbolplot(A; color=:black); title("Butterfly")
+clf(); qlplot(transpose(A); branch=branch(2), x=range(-3,3; length=100), y=range(-3,3;length=100)); symbolplot(A; color=:black); title("Butterfly")

src/blockbanded/infblocktridiagonal.jl

@@ -38,6 +38,10 @@ function sizes_from_blocks(A::AbstractMatrix, ::Tuple{OneTo{Int}, OneToInf{Int}}
     BlockSizes((vcat(1, 1 .+ cumsum(R)), Vcat(1, 1 .+ cumsum(C))))
 end
 
+function sizes_from_blocks(A::Diagonal, ::NTuple{2,OneToInf{Int}}) 
+    sz = size.(A.diag, 1), size.(A.diag,2)
+    BlockSizes(Vcat.(1,(c -> 1 .+ c).(cumsum.(sz))))
+end
 
 function sizes_from_blocks(A::Tridiagonal, ::NTuple{2,OneToInf{Int}}) 
     sz = size.(A.d, 1), size.(A.d,2)