Wie interpoliert man in einen Julia "for" Ausdruck?

Question

Ich schrieb ein Makro @vcomp (Vektor Verständnis) basierend auf Python Listen-Comprehensions mit einer bedingten Klausel, um Elemente in einer prägnanten Art und Weise zu filtern.

macro vcomp(comprehension::Expr, when::Symbol, condition) 
    comp_head, comp_args = comprehension.head, comprehension.args 
    comp_head ∉ [:comprehension, :typed_comprehension] && error("@vcomp not a comprehension") 
    when ≠ :when && error("@vcomp expected `when`, got: `$when`") 
    T = comp_head == :typed_comprehension ? comp_args[1] : nothing 
    if VERSION < v"0.5-" 
     element = comp_head == :comprehension ? comp_args[1] : comp_args[2] 
     sequence = comp_head == :comprehension ? comp_args[2] : comp_args[3] 
    else 
     element = comp_head == :comprehension ? comp_args[1].args[1] : comp_args[2].args[1] 
     sequence = comp_head == :comprehension ? comp_args[1].args[2] : comp_args[2].args[2] 
    end 
    result = T ≠ nothing ? :($T[]) : :([]) 
    block = Expr(:let, Expr(:block, 
       Expr(:(=), :res, result), 
       Expr(:for, sequence, 
        Expr(:if, condition, 
         Expr(:call, :push!, :res, element))), 
       :res)) 
    return esc(block) 
end 

wie folgt verwendet:

julia> @vcomp Int[i^3 for i in 1:10] when i % 2 == 0 
5-element Array{Int64,1}: 
    8 
    64 
    216 
    512 
1000 

auf das sich diese erweitern:

julia> macroexpand(:(@vcomp Int[i^3 for i in 1:15] when i % 2 == 0)) 
:(let 
     res = Int[] 
     for i = 1:15 
      if i % 2 == 0 
       push!(res,i^3) 
      end 
     end 
     res 
    end) 

Ich hatte erwartet, der Lage sein, dies zu schreiben block wie:

block = quote 
    let 
     res = $result 
     for $sequence 
      if $condition 
       push!(res, $element) 
      end 
     end 
     res 
    end 
end 

Welche der folgenden Fehler gibt:

• ERROR: syntax: invalid iteration specification

Statt der Art und Weise kam ich mit:

block = Expr(:let, Expr(:block, 
      Expr(:(=), :res, result), 
      Expr(:for, sequence, 
       Expr(:if, condition, 
        Expr(:call, :push!, :res, element))), 
      :res)) 

aber ich war in der Lage, es zu tun Expr(:for, ...) direkt, wie oben gezeigt, und so weit ich das verstehe, ist das ein Parserfehler (ist das ein Bug?). Ich habe auch nicht in der Lage gewesen, Beispiele für diese Art von Interpolation zu finden, das ist, was ich versucht habe:

julia> ex₁ = :(i in 1:10) 
:($(Expr(:in, :i, :(1:10)))) 

julia> ex₂ = :(i = 1:10) 
:(i = 1:10) 

julia> quote 
      for $ex₁ 
ERROR: syntax: invalid iteration specification 

julia> quote 
      for $ex₂ 
ERROR: syntax: invalid iteration specification 

Construct ganzen Ausdruck und prüfen:

julia> ex₃ = quote 
      for i in 1:10 
       print(i) 
      end 
     end 
quote # none, line 2: 
    for i = 1:10 # none, line 3: 
     print(i) 
    end 
end 

julia> ex₃.args 
2-element Array{Any,1}: 
:(# none, line 2:) 
:(for i = 1:10 # none, line 3: 
     print(i) 
    end) 

julia> ex₃.args[2].args 
2-element Array{Any,1}: 
:(i = 1:10) 
quote # none, line 3: 
    print(i) 
end 

julia> ex₃.args[2].args[1] 
:(i = 1:10) 

julia> ex₃.args[2].args[1] == ex₂ # what's the difference then? 
true 

Das funktioniert aber weniger lesbar:

julia> ex₄ = Expr(:for, ex₁, :(print(i))) 
:(for $(Expr(:in, :i, :(1:10))) 
     print(i) 
    end) 

julia> ex₅ = Expr(:for, ex₂, :(print(i))) 
:(for i = 1:10 
     print(i) 
    end) 

julia> eval(ex₃) 
12345678910 
julia> eval(ex₄) 
12345678910  
julia> eval(ex₅) 
12345678910  

Gibt es eine Möglichkeit, die eher knappe Syntax stattdessen verwenden? Ich finde die aktuelle Implementierung schwierig zu lesen und zu begründen im Vergleich zu dem, was ich zu schreiben erwartet hatte.

Answer 1

Zunächst einmal glaube ich, dass Nachforschungen mit Wachen kommen Julia (in v0.5?).

Um Ihre Frage zu beantworten: Der Parser möchte in der Lage sein zu überprüfen, dass seine Eingabe syntaktisch korrekt ist, ohne in den tatsächlichen Wert zu interpolieren. Versuchen Sie zB

x, y = :i, :(1:10) 
quote 
    for $x = $y 
    end 
end 

Jetzt kann der Parser die relevanten Teile der Syntax erkennen. (Und Sie sollten die gleiche AST erhalten, wenn Sie stattdessen for $x in $y verwenden.)