Избегание выражения в макросе-сгенерированном макросе в Юлии - PullRequest
Новая
       20

Избегание выражения в макросе-сгенерированном макросе в Юлии

2 голосов
/ 27 марта 2020

Я хочу написать макрос в Julia, который генерирует количество макросов 

@register_attribute foo

должен генерировать макросы 

@set_foo
@get_foo

, которые, в свою очередь, должны быть определены как 

@set_foo x 5 == set_attr!(x, :foo, 5)

Мне удалось получить первую часть этого штрафа, однако @macroexpand показывает, что внутренняя переменная x не экранирована должным образом, в результате UndefVarErrors, если я вызываю метод с локальной переменной . 

@macroexpand @set_foo x 5 =
    :(Main.set_attr!(Main.x, (Symbol)("foo"), 5))

, тогда как я хочу 

@macroexpand @set_foo x 5 =
    :(set_attr!(x, (Symbol)("foo"), 5))

Мой код 

macro register_attribute(name)
    setn = Symbol("set_", string(name))
    getn = Symbol("get_", string(name))
    arg = gensym()
    arg2 = gensym()
    nsym = string(name)
    return Expr(:block, 
    esc(:(
        macro $(setn)($arg, $arg2)
            :(set_attr!($$arg, $$Symbol($$nsym), $$arg2)) 
        end)
        ),
    esc(:(
        macro $(getn)($arg, $arg2)
            :(get_attr!($$arg, $$Symbol($$nsym), $$arg2)) 
        end)
        ))
    e2
end

1 Ответ

4 голосов
/ 27 марта 2020

Вот как бы я подошел ко всей этой проблеме: 

# for purely generative tasks like this, where the generated code is
# (presumably?) destined to be evalled at top level, I prefer using @eval
for attr in [:foo, :bar]
    get_attr_name = Symbol("get_", attr)
    set_attr_name = Symbol("set_", attr)
    quoted_attr   = QuoteNode(attr)

    @eval begin
        $get_attr_name(x)      = getproperty( x, $quoted_attr)
        $set_attr_name(x, val) = setproperty!(x, $quoted_attr, val)
    end
end

# should generate code like:
#   get_foo(x) = getproperty(x, :foo)

julia> nt = (foo=1, bar=2);

julia> get_foo(nt)
1

julia> get_bar(nt)
2

# Let's test in local scope too
julia> let nt2 = (foo=42,)
           get_foo(nt2)
       end
42
...