Adjoint/Transpose conform block array interface

Author: dlfivefifty

src/BlockArrays.jl

@@ -27,7 +27,7 @@ import Base: @propagate_inbounds, Array, to_indices, to_index,
 
 import Base: (:), IteratorSize, iterate, axes1
 import Base.Broadcast: broadcasted, DefaultArrayStyle
-import LinearAlgebra: lmul!, rmul!, AbstractTriangular
+import LinearAlgebra: lmul!, rmul!, AbstractTriangular, HermOrSym, AdjOrTrans
 
 
 include("abstractblockarray.jl")

src/blockarrayinterface.jl

@@ -2,8 +2,9 @@
 blocksizes(A::AbstractTriangular) = blocksizes(parent(A))
 
 
-blocksizes(A::Symmetric) = blocksizes(parent(A))
-blocksizes(A::Hermitian) = blocksizes(parent(A))
+blocksizes(A::HermOrSym) = blocksizes(parent(A))
+blocksizes(A::AdjOrTrans{<:Any,<:AbstractMatrix}) = BlockSizes(reverse(cumulsizes(blocksizes(parent(A)))))
+blocksizes(A::AdjOrTrans{<:Any,<:AbstractVector}) = BlockSizes(([1,2],cumulsizes(blocksizes(parent(A)))[1]))
 
 Base.print_matrix_row(io::IO,
         X::Union{AbstractTriangular{<:Any,<:AbstractBlockMatrix},

test/test_blockarrayinterface.jl

@@ -39,3 +39,19 @@ end
     @test Hermitian(A)[Block(2,3)] == A[2:3,4:6]
     @test Hermitian(A)[Block(3,2)] == A[2:3,4:6]'
 end
+
+@testset "Adjoint/Transpose block arrays" begin
+    A = PseudoBlockArray{ComplexF64}(undef, (1:4), (2:5))
+    A .= randn.() .+ randn.().*im
+
+    @test blocksizes(A') == BlockArrays.BlockSizes(2:5, 1:4)
+    @test blocksizes(Transpose(A)) == BlockArrays.BlockSizes(2:5, 1:4)
+
+    @test A'[Block(2,2)] == A[Block(2,2)]' == A[2:3,3:5]'
+    @test transpose(A)[Block(2,2)] == transpose(A[2:3,3:5])
+    @test A'[Block(2,3)] == A[Block(3,2)]'
+    @test transpose(A)[Block(2,3)] == transpose(A[Block(3,2)])
+
+    @test BlockArray(A') == A'
+    @test BlockArray(transpose(A)) == transpose(A)
+end