Added donorview and haloview functions

Project.toml

@@ -1,7 +1,7 @@
 name = "BlockHaloArrays"
 uuid = "8029ca6b-11ad-4a59-88a2-2e6eee4ef8a2"
 authors = ["Sam Miller <[email protected]> and contributors"]
-version = "0.3.0"
+version = "0.3.1"
 
 [deps]
 EllipsisNotation = "da5c29d0-fa7d-589e-88eb-ea29b0a81949"

src/BlockHaloArrays.jl

@@ -13,6 +13,7 @@ export flatten, repartition!, updatehalo!, updateblockhalo!
 export globalindices, nblocks
 export domainview
 export onboundary
+export donorview, haloview
 
 abstract type AbstractBlockHaloArray end
 
@@ -309,4 +310,10 @@ function domainview(A::BlockHaloArray, blockid::Integer)
     return @views A.blocks[blockid][.., Tuple(idx_range_vec)...]
 end
 
+"""Get the halo region at a particular location, e.g. `:ilo` for block `blockid`"""
+haloview(A::BlockHaloArray, blockid::Integer, location::Symbol) = A._halo_views[blockid][location]
+
+"""Get the domain donor region at a particular location, e.g. `:ilo` for block `blockid`"""
+donorview(A::BlockHaloArray, blockid::Integer, location::Symbol) = A._donor_views[blockid][location]
+
 end # module

test/unit/test_util_funcs.jl

@@ -0,0 +1,30 @@
+@testitem "halo/donorview" begin
+    include("common.jl")
+    dims = (50, 50)
+    nhalo = 2
+    nblocks = 4
+    A = BlockHaloArray(dims, nhalo, nblocks)
+
+    @test size(donorview(A, 1, :ilo)) == (2, 25)
+    @test size(haloview(A, 1, :ilo)) == (2, 25)
+    @test size(donorview(A, 1, :jhi)) == (25, 2)
+    @test size(haloview(A, 1, :jhi)) == (25, 2)
+end
+
+@testitem "nblocks" begin
+    include("common.jl")
+    A = BlockHaloArray((50,50), 2, 4)
+    @test nblocks(A) == 4
+end
+
+@testitem "eltype" begin
+    include("common.jl")
+    A = BlockHaloArray((50,50), 2, 4, T=Int32)
+    @test eltype(A) == Int32
+end
+
+@testitem "size" begin
+    include("common.jl")
+    A = BlockHaloArray((50,50), 2, 4, T=Int32)
+    @test size(A) == (50, 50)
+end