up

Author: exAClior

example/floquetify.jl

@@ -6,29 +6,40 @@ using ZXCalculus.Utils: Parameter
 using WGLMakie, GraphMakie
 
 function make_meas_mg(weight::T) where {T<:Integer}
-    spiders_mg = Multigraph(weight + 1)
+    spiders_mg = Multigraph(weight * 3 + 1)
+
+    for i in 1:weight
+        add_edge!(spiders_mg, 2 * (i - 1) + 1, 2 * weight + i)
+        add_edge!(spiders_mg, 2 * (i - 1) + 2, 2 * weight + i)
+    end
+
     for i in 1:weight
-        add_edge!(spiders_mg, i, weight + 1, 1)
+        add_edge!(spiders_mg, 2 * weight + i, 3 * weight + 1)
     end
+
     return spiders_mg
 end
 
-function make_meas_zxwd(::Type{P}, mg::Multigraph{T}, meas_type::Symbol) where {T<:Integer, P}
-	weight = nv(mg) - 1
-    st_vec = [[(meas_type == :X) ? ZXW.Z(zero(P)) : ZXW.X(zero(P)) for _ in 1:weight]..., (meas_type == :X) ? ZXW.X(zero(P)) : ZXW.Z(zero(P))]
+function make_meas_zxwd(::Type{P}, mg::Multigraph{T}, meas_type::Symbol) where {T<:Integer,P}
+    weight = (nv(mg) - 1) // 3
+    st_vec = [[isodd(i) ? ZXW.Input(i ÷ 2 + 1) : ZXW.Output(i ÷ 2) for i in 1:2*weight]..., [(meas_type == :X) ? ZXW.Z(zero(P)) : ZXW.X(zero(P)) for _ in 1:weight]..., (meas_type == :X) ? ZXW.X(zero(P)) : ZXW.Z(zero(P))]
+    @show st_vec
     return ZXWDiagram(mg, st_vec)
 end
 
 weight = 4
 meas_mg = make_meas_mg(weight)
 
+xstab_meas_zxwd = make_meas_zxwd(Parameter, meas_mg, :X)
+zstab_meas_zxwd = make_meas_zxwd(Parameter, meas_mg, :Z)
+
+ZXCalculus.ZXW.plot(xstab_meas_zxwd)
+ZXCalculus.ZXW.plot(zstab_meas_zxwd)
 
-xstab_meas_zxd = make_meas_zxwd(Parameter, meas_mg, :X)
-zstab_meas_zxd = make_meas_zxwd(Parameter, meas_mg, :Z)
 
-ZXCalculus.ZXW.plot(xstab_meas_zxd)
+concat_zxwd = ZXW.concat!(copy(xstab_meas_zxwd), copy(zstab_meas_zxwd))
 
-typeof(xstab_meas_zxd)
+ZXCalculus.ZXW.plot(concat_zxwd)
 
 # Convert requirement checking into linear programming
 # Req1: even number of green/red spiders with degree 1 respectively

ext/ZXWCalculusExt.jl

@@ -5,32 +5,36 @@ using WGLMakie.Colors
 using MLStyle
 using ZXCalculus: ZXW
 using ZXCalculus, ZXCalculus.ZXW
-using ZXCalculus.ZXW: Z,X
+using ZXCalculus.ZXW: Z, X, Input, Output, spider_type
 using ZXCalculus.ZX.Graphs
 
-function edge_click_action(idx, args...)
-    red_green_blk = [RGB(1.,0.,0.), RGB(0.,1.,0.), RGB(0.,0.,0.)]
-    which_color = findfirst(x -> x == p.edge_color[][idx], red_green_blk)
-    p.edge_color[][idx] = red_green_blk[mod1(which_color+1,length(red_green_blk))]
-    p.edge_color[] = p.edge_color[]
-end
 
 function ZXCalculus.ZXW.plot(zxwd::ZXWDiagram{T,P}; kwargs...) where {T,P}
     g = zxwd.mg
     f, ax, p = graphplot(g,
         edge_width=[2.0 for i in 1:ne(g)],
         edge_color=[colorant"black" for i in 1:ne(g)],
-        node_size=[10 for i in 1:nv(g)],
+        node_size=[15 for i in 1:nv(g)],
         node_color=[
             @match spider_type(zxwd, i) begin
                 Z(p) => colorant"red"
                 X(p) => colorant"green"
+                Input => colorant"orange"
+                Output => colorant"orange"
                 _ => colorant"yellow"
             end for i in 1:nv(g)])
 
     hidedecorations!(ax)
     hidespines!(ax)
     deregister_interaction!(ax, :rectanglezoom)
+
+    function edge_click_action(idx, args...)
+        red_green_blk = [RGB(1.0, 0.0, 0.0), RGB(0.0, 1.0, 0.0), RGB(0.0, 0.0, 0.0)]
+        which_color = findfirst(x -> x == p.edge_color[][idx], red_green_blk)
+        p.edge_color[][idx] = red_green_blk[mod1(which_color + 1, length(red_green_blk))]
+        p.edge_color[] = p.edge_color[]
+    end
+
     register_interaction!(ax, :nhover, NodeHoverHighlight(p))
     register_interaction!(ax, :ehover, EdgeHoverHighlight(p))
     register_interaction!(ax, :ndrag, NodeDrag(p))

src/ZXW/utils.jl

@@ -498,6 +498,7 @@ function concat!(d1::ZXWDiagram{T,P}, d2::ZXWDiagram{T,P}) where {T,P}
     for i = 1:nout(d1)
         out_idx = get_output_idx(d1, i)
         # output spiders cannot be connected to multiple vertices or with multiedge
+        @show out_idx
         prior_vtx = neighbors(d1, out_idx)[1]
         rem_edge!(d1, out_idx, prior_vtx)
         # d2 input vtx idx is mapped to the vtx prior to d1 output