Merge pull request #517 from nzy1997/zy/label-color

Plot with colored label

lib/YaoBlocks/test/primitive/time_evolution.jl

@@ -108,18 +108,18 @@ end
  mpb = mat(pb)
  allpass = true
  for i=basis(pb), j=basis(pb)
- allpass &= pb[i, j] ≈ mpb[Int(i)+1, Int(j)+1]
+ allpass &= isapprox(pb[i, j], mpb[Int(i)+1, Int(j)+1]; atol=1e-6, rtol=1e-6)
  end
  @test allpass
 
  allpass = true
  for j=basis(pb)
- allpass &= vec(pb[:, j]) ≈ mpb[:, Int(j)+1]
- allpass &= vec(pb[j,:]) ≈ mpb[Int(j)+1,:]
- allpass &= vec(pb[:, EntryTable([j], [1.0+0im])]) ≈ mpb[:, Int(j)+1]
- allpass &= vec(pb[EntryTable([j], [1.0+0im]),:]) ≈ mpb[Int(j)+1,:]
+ allpass &= isapprox(vec(pb[:, j]), mpb[:, Int(j)+1]; atol=1e-6, rtol=1e-6)
+ allpass &= isapprox(vec(pb[j,:]), mpb[Int(j)+1,:]; atol=1e-6, rtol=1e-6)
+ allpass &= isapprox(vec(pb[:, EntryTable([j], [1.0+0im])]), mpb[:, Int(j)+1]; atol=1e-6, rtol=1e-6)
+ allpass &= isapprox(vec(pb[EntryTable([j], [1.0+0im]),:]), mpb[Int(j)+1,:]; atol=1e-6, rtol=1e-6)
  allpass &= isclean(pb[:,j])
  end
  @test allpass
  end
-end
+end

lib/YaoPlots/src/helperblock.jl

@@ -9,11 +9,12 @@ A marker to mark a circuit applying on a continous block for better plotting.
 struct LabelBlock{BT<:AbstractBlock,D} <: TagBlock{BT,D}
  content::BT
  name::String
+ color::String
 end
 
 YaoBlocks.content(cb::LabelBlock) = cb.content
-function LabelBlock(x::BT, name::String) where {D,BT<:AbstractBlock{D}}
- LabelBlock{BT,D}(x, name)
+function LabelBlock(x::BT, name::String, color::String) where {D,BT<:AbstractBlock{D}}
+ LabelBlock{BT,D}(x, name, color)
 end
 
 function is_continuous_chunk(x)
@@ -24,14 +25,14 @@ end
 YaoBlocks.PropertyTrait(::LabelBlock) = YaoBlocks.PreserveAll()
 YaoBlocks.mat(::Type{T}, blk::LabelBlock) where {T} = mat(T, content(blk))
 YaoBlocks.unsafe_apply!(reg::YaoBlocks.AbstractRegister, blk::LabelBlock) = YaoBlocks.unsafe_apply!(reg, content(blk))
-YaoBlocks.chsubblocks(blk::LabelBlock, target::AbstractBlock) = LabelBlock(target, blk.name)
+YaoBlocks.chsubblocks(blk::LabelBlock, target::AbstractBlock) = LabelBlock(target, blk.name, blk.color)
 
-Base.adjoint(x::LabelBlock) = LabelBlock(adjoint(content(x)), endswith(x.name, "†") ? x.name[1:end-1] : x.name*"†")
-Base.copy(x::LabelBlock) = LabelBlock(copy(content(x)), x.name)
+Base.adjoint(x::LabelBlock) = LabelBlock(adjoint(content(x)), endswith(x.name, "†") ? x.name[1:end-1] : x.name*"†", x.color)
+Base.copy(x::LabelBlock) = LabelBlock(copy(content(x)), x.name, x.color)
 YaoBlocks.Optimise.to_basictypes(block::LabelBlock) = block
 
 export addlabel
-addlabel(b::AbstractBlock, str::String) = LabelBlock(b, str)
+addlabel(b::AbstractBlock; name=string(b), color="transparent") = LabelBlock(b, name, color)
 
 # to fix issue 
 function YaoBlocks.print_tree(

lib/YaoPlots/src/vizcircuit.jl

@@ -360,14 +360,20 @@ function draw!(c::CircuitGrid, cb::ControlBlock{GT,C}, address, controls) where
  draw!(c, cb.content, locs, [controls..., mycontrols...])
 end
 
-for B in [:LabelBlock, :GeneralMatrixBlock, :Add]
+for B in [:GeneralMatrixBlock, :Add]
  @eval function draw!(c::CircuitGrid, cb::$B, address, controls)
  length(address) == 0 && return
- #is_continuous_chunk(address) || error("address not continuous in a block marked as continous.")
  _draw!(c, [controls..., (address, c.gatestyles.g, string(cb))])
  end
 end
 
+function draw!(c::CircuitGrid, cb::LabelBlock, address, controls)
+ length(address) == 0 && return
+ CircuitStyles.gate_bgcolor[], temp = cb.color, CircuitStyles.gate_bgcolor[]
+ _draw!(c, [controls..., (address, c.gatestyles.g, string(cb))])
+ CircuitStyles.gate_bgcolor[] = temp
+end
+
 # [:KronBlock, :RepeatedBlock, :CachedBlock, :Subroutine, :(YaoBlocks.AD.NoParams)]
 function draw!(c::CircuitGrid, p::CompositeBlock, address, controls)
  barrier_style = CircuitStyles.barrier_for_chain[]

lib/YaoPlots/test/helperblock.jl

@@ -4,7 +4,7 @@ using Test
 
 @testset "LabelBlock" begin
  x = put(5, (2,3)=>matblock(rand_unitary(4)))
- cb = LabelBlock(x, "x")
+ cb = LabelBlock(x, "x", "red")
  @test mat(copy(cb)) == mat(cb)
  @test isunitary(cb)
  @test ishermitian(cb) == ishermitian(x)
@@ -21,8 +21,8 @@ using Test
  @test YaoPlots.is_continuous_chunk([1,2,4]) == false
  @test YaoPlots.is_continuous_chunk([3,2,4]) == true
 
- c1 = chain(5, [put(5, (2,3)=>addlabel(SWAP, "SWAP")), put(5, 2=>addlabel(I2, "id")), put(5, 2=>addlabel(X, "X")), control(5, (5,3), (2,4,1)=>put(3, (1,3)=>addlabel(SWAP, "SWAP")))])
- c2 = chain(5, [put(5, (2,3)=>addlabel(SWAP, "SWAP")), put(5, 2=>addlabel(I2, "id")), put(5, 2=>addlabel(X, "X")), control(5, (5,3), (2,4,1)=>put(3, (1,2)=>addlabel(SWAP, "SWAP")))])
+ c1 = chain(5, [put(5, (2,3)=>addlabel(SWAP; name="SWAP")), put(5, 2=>addlabel(I2; name="id")), put(5, 2=>addlabel(X; name="X")), control(5, (5,3), (2,4,1)=>put(3, (1,3)=>addlabel(SWAP; name="SWAP")))])
+ c2 = chain(5, [put(5, (2,3)=>addlabel(SWAP; name="SWAP")), put(5, 2=>addlabel(I2; name="id")), put(5, 2=>addlabel(X; name="X")), control(5, (5,3), (2,4,1)=>put(3, (1,2)=>addlabel(SWAP; name="SWAP")))])
 
  @test vizcircuit(c1) isa Drawing
  @test vizcircuit(c2) isa Drawing