Question

我希望能够为用户定义的类型创建一个调度，它基本上会执行一个就地复制。但是，我想以类型稳定的方式进行，因此我想避免直接使用getfield，而是尝试使用生成的函数。是否可以使用类似

的类型

type UserType{T}
  x::Vector{T}
  y::Vector{T}
  z::T
end

生成一些功能

recursivecopy!(A::UserType,B::UserType)
  # Do it for x
  if typeof(A.x) <: AbstractArray
    recursivecopy!(A.x,B.x)
  else
    A.x = B.x
  end
  # Now for y
  if typeof(A.y) <: AbstractArray
    recursivecopy!(A.y,B.y)
  else
    A.y = B.y
  end
  # Now for z
  if typeof(A.z) <: AbstractArray
    recursivecopy!(A.z,B.z)
  else
    A.z = B.z
  end
end

RecursiveArrayTools.jl中的recursivecopy!使得这个句柄嵌套（Vector{Vector}）类型很好，但唯一的问题是我不知道用户将提前拥有的字段，只是在编译时 - 调用此函数的时间。听起来像生成函数的工作，但我不太确定如何生成它。

Answer 1

您不需要向后弯腰以避免getfield和setfield。朱莉娅可以推断他们就好了。当Julia无法弄清楚它正在访问哪个字段时...就像在for循环中一样，麻烦来了。

因此，生成的函数需要做的唯一特殊事情是使用拼接到getfield的常量值有效地展开循环：

julia> immutable A
           x::Int
           y::Float64
       end

julia> @generated function f(x)
           args = [:(getfield(x, $i)) for i=1:nfields(x)]
           :(tuple($(args...)))
       end
f (generic function with 1 method)

julia> f(A(1,2.4))
(1,2.4)

julia> @code_warntype f(A(1,2.4))
Variables:
  #self#::#f
  x::A

Body:
  begin  # line 2:
      return (Main.tuple)((Main.getfield)(x::A,1)::Int64,(Main.getfield)(x::A,2)::Float64)::Tuple{Int64,Float64}
  end::Tuple{Int64,Float64}

就像你可以将多个参数拼接到函数调用中一样，你也可以直接将多个表达式拼接到函数体中。

julia> type B
           x::Int
           y::Float64
       end
julia> @generated function f!{T}(dest::T, src::T)
           assignments = [:(setfield!(dest, $i, getfield(src, $i))) for i=1:nfields(T)]
           :($(assignments...); dest)
       end
f! (generic function with 1 method)

julia> f!(B(0,0), B(1, 2.4))
B(1,2.4)

julia> @code_warntype f!(B(0,0), B(1, 2.4))
Variables:
  #self#::#f!
  dest::B
  src::B

Body:
  begin  # line 2:
      (Main.setfield!)(dest::B,1,(Main.getfield)(src::B,1)::Int64)::Int64
      (Main.setfield!)(dest::B,2,(Main.getfield)(src::B,2)::Float64)::Float64
      return dest::B
  end::B

当然，你可以使这种理解的主体像你一样复杂。这有效地成为for循环的内部。将数组映射到函数体中会为您展开。

使用生成的函数生成类型稳定的`getfield`调用

1 个答案: