From 7f94ae9757960d8f519bec7e61cbe62b9d974406 Mon Sep 17 00:00:00 2001
From: Jishnu Bhattacharya <jishnub.github@gmail.com>
Date: Tue, 2 Apr 2024 18:33:27 +0530
Subject: [PATCH] Vector indexing a PseudoBlockArray of a FillArray (#378)

---
 Project.toml             | 2 +-
 src/pseudo_blockarray.jl | 6 ++++++
 test/test_blockarrays.jl | 7 +++++++
 3 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/Project.toml b/Project.toml
index 6ad68474..ae6532dd 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,6 +1,6 @@
 name = "BlockArrays"
 uuid = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"
-version = "0.16.43"
+version = "0.16.44-dev"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"
diff --git a/src/pseudo_blockarray.jl b/src/pseudo_blockarray.jl
index a635822a..81f6a250 100644
--- a/src/pseudo_blockarray.jl
+++ b/src/pseudo_blockarray.jl
@@ -356,3 +356,9 @@ Base.replace_in_print_matrix(f::PseudoBlockVecOrMat, i::Integer, j::Integer, s::
 
 
 LinearAlgebra.norm(A::PseudoBlockArray, p::Real=2) = norm(A.blocks, p)
+
+###########################
+# FillArrays interface #
+###########################
+
+FillArrays.getindex_value(P::PseudoBlockArray) = FillArrays.getindex_value(P.blocks)
diff --git a/test/test_blockarrays.jl b/test/test_blockarrays.jl
index 66d7e6de..92e20c1d 100644
--- a/test/test_blockarrays.jl
+++ b/test/test_blockarrays.jl
@@ -555,6 +555,13 @@ end
         @test BLAS.gemv!('N', 2.0, A, x, 0.0, y) ≈ 2A*x
     end
 
+    @testset "FillArrays interface" begin
+        P = PseudoBlockArray(Fill(3,4,4), [1,3], [1,3])
+        @test P[1:3, 2:3] === Fill(3,3,2)
+        @test P[1:3, 1] == Fill(3,3)
+        @test P[2, 1:3] == Fill(3,3)
+    end
+
     @testset "lmul!/rmul!" begin
         A = PseudoBlockArray{Float64}(undef, 1:3)
         @test fill!(A, NaN) === A