quickfix of QFT

Author: GiggleLiu

src/QFT.jl

@@ -11,8 +11,8 @@ QFTCircuit(n::Int) = chain(n, CRot(n, i) for i = 1:n)
 struct QFTBlock{N} <: PrimitiveBlock{N,ComplexF64} end
 mat(q::QFTBlock{N}) where N = applymatrix(q)
 
-apply!(reg::DefaultRegister{B}, ::QFTBlock) where B = (reg.state = ifft!(invorder_firstdim(reg |> state), 1)*sqrt(1<<nqubits(reg)); reg)
-apply!(reg::DefaultRegister{B}, ::Daggered{N, T, <:QFTBlock}) where {B,N,T} = (reg.state = invorder_firstdim(fft!(reg|>state, 1)/sqrt(1<<nqubits(reg))); reg)
+apply!(reg::DefaultRegister{B}, ::QFTBlock) where B = (reg.state = ifft!(invorder_firstdim(reg |> state), 1)*sqrt(1<<nactive(reg)); reg)
+apply!(reg::DefaultRegister{B}, ::Daggered{N, T, <:QFTBlock}) where {B,N,T} = (reg.state = invorder_firstdim(fft!(reg|>state, 1)/sqrt(1<<nactive(reg))); reg)
 
 # traits
 ishermitian(q::QFTBlock{N}) where N = N==1

src/QuAlgorithmZoo.jl

@@ -5,7 +5,7 @@ using Yao
 using Yao.Intrinsics
 using Yao.Registers
 using Yao.Blocks
-import Yao.Blocks: mat, dispatch!, nparameters, parameters, cache_key, print_block, _make_rot_mat, apply!
+import Yao.Blocks: mat, dispatch!, nparameters, parameters, cache_key, print_block, _make_rot_mat, apply!, PrimitiveBlock
 import Base: ==, copy, hash
 import Yao.Intrinsics: ishermitian, isreflexive, isunitary
 

test/QFT.jl

@@ -44,4 +44,7 @@ end
 
     # test fft
     @test apply!(copy(reg), qft) ≈ apply!(copy(reg), qftblock)
+
+    # regression test for nactive
+    @test apply!(focus!(copy(reg), 1:3), QFTBlock{3}()) |> isnormalized
 end