added blockiszero

Author: lostella

src/utilities/block.jl

@@ -14,7 +14,8 @@ export RealOrComplex,
        blockset!,
        blockvecdot,
        blockzeros,
-       blockaxpy!
+       blockaxpy!,
+       blockiszero
 
 
 const RealOrComplex{R} = Union{R, Complex{R}}
@@ -65,6 +66,9 @@ blockzeros(a::AbstractArray) = zeros(a)
 blockaxpy!(z::Tuple, x, alpha::Real, y::Tuple) = blockaxpy!.(z, x, alpha, y)
 blockaxpy!(z::AbstractArray, x, alpha::Real, y::AbstractArray) = (z .= x .+ alpha.*y)
 
+blockiszero(x::AbstractArray) = iszero(x)
+blockiszero(x::Tuple) = all(iszero.(x))
+
 # Define broadcast
 
 import Base: broadcast!

test/test_block.jl

@@ -32,15 +32,15 @@ x2b = (randn(2), randn(3)+im.*randn(3), randn(2,3))
 
 y = blocksimilar(x)
 yb = blocksimilar(xb)
-blockcopy!(y,x) 
+blockcopy!(y,x)
 blockcopy!(yb,xb)
 
 @test y == x
 @test yb == xb
 
 y = blocksimilar(x)
 yb = blocksimilar(xb)
-blockset!(y,x) 
+blockset!(y,x)
 blockset!(yb,xb)
 
 @test y == x
@@ -72,8 +72,18 @@ blockaxpy!(yb,xb,2,x2b)
 @test y == x .+ 2.*x2
 @test yb == (xb[1] .+ 2.*x2b[1], xb[2] .+ 2.*x2b[2], xb[3] .+ 2.*x2b[3])
 
+x = rand(Float64, 5)
+y = (x, x)
+@test blockiszero(x) == blockiszero(y) == false
 
+x = rand(Complex{Float64}, 5)
+y = (x, x)
+@test blockiszero(x) == blockiszero(y) == false
 
+x = zeros(Float64, 5)
+y = (x, x)
+@test blockiszero(x) == blockiszero(y) == true
 
-
-
+x = zeros(Complex{Float64}, 5)
+y = (x, x)
+@test blockiszero(x) == blockiszero(y) == true