Julia中如何向调用宏的方法传递变量？

问题背景 #

在函数内部调用NCDatasets包的@select宏，参数由方法传入，直接调用宏时功能正常，但在函数内调用时，lat和lon变量无法被宏正确识别展开，触发世界年龄相关的方法错误。

出错代码 #

function data_for_lat_lon(ds, region, lat_lon_pair)
    println("have latlon piar ", lat_lon_pair)
    lat = lat_lon_pair[1]
    lon = lat_lon_pair[2]
    data = []
    if(ArchGDAL.contains(region[1], ArchGDAL.createpoint(lon, lat)))
        println(lat, " ", lon)
        # 此处调用会失败
        single_lat_lon = NCDatasets.@select(ds, latitude==$lat && longitude==$lon)
        for (varname, var) in single_lat_lon
            if varname in ["latitude", "longitude", "time"]
                continue
            end
            push!(var_names, varname)
            push!(data, Array[single_lat_lon[varname]][1][:])
        end

        return reduce(hcat, data)'
    end
end

错误信息及栈跟踪 #

MethodError: no method matching (::NCDatasets.var"#154#155")(::Float64)
The applicable method may be too new: running in world age 32645, while current world is 32646.
Closest candidates are:
  (::NCDatasets.var"#154#155")(::Any) at none:0 (method too new to be called from this world context.)

Stacktrace:
  [1] _broadcast_getindex_evalf
    @ .\broadcast.jl:670 [inlined]
  [2] _broadcast_getindex
    @ .\broadcast.jl:643 [inlined]
  [3] getindex
    @ .\broadcast.jl:597 [inlined]
  [4] copy
    @ .\broadcast.jl:899 [inlined]
  [5] materialize
    @ .\broadcast.jl:860 [inlined]
  [6] findall(testf::NCDatasets.var"#154#155", A::Vector{Float64})
    @ Base .\array.jl:2311
  [7] macro expansion
    @ C:\Users\scott\.julia\packages\NCDatasets\sLdiM\src\select.jl:242 [inlined]
  [8] data_for_lat_lon(ds::NCDatasets.MFDataset{DeferDataset, 1, String, NCDatasets.DeferAttributes, NCDatasets.DeferDimensions, NCDatasets.DeferGroups}, region::DataFrameRow{DataFrame, DataFrames.Index}, lat_lon_pair::Tuple{Float64, Float64})
    @ Main .\In[46]:8
  [9] top-level scope
    @ .\In[50]:3
 [10] eval
    @ .\boot.jl:368 [inlined]
 [11] include_string(mapexpr::typeof(REPL.softscope), mod::Module, code::String, filename::String)
    @ Base .\loading.jl:1428

问题分析 #

错误根源在于宏的卫生性机制与函数编译时机冲突：

• 宏展开时，$lat和$lon的变量捕获方式在函数局部作用域内出现上下文不匹配，导致生成的匿名函数无法正确访问函数内的局部变量。
• 世界年龄错误说明宏动态生成的新方法，在函数编译阶段还未被当前Julia会话识别，属于宏展开与函数编译的时机错位问题。

解决方法 #

方法1：用@eval动态展开宏调用 #

通过@eval强制在运行时动态生成宏调用代码，确保变量值直接嵌入宏逻辑中：

function data_for_lat_lon(ds, region, lat_lon_pair)
    println("have latlon piar ", lat_lon_pair)
    lat = lat_lon_pair[1]
    lon = lat_lon_pair[2]
    data = []
    if(ArchGDAL.contains(region[1], ArchGDAL.createpoint(lon, lat)))
        println(lat, " ", lon)
        # 使用@eval动态展开变量与宏
        single_lat_lon = @eval NCDatasets.@select($ds, latitude==$lat && longitude==$lon)
        for (varname, var) in single_lat_lon
            if varname in ["latitude", "longitude", "time"]
                continue
            end
            push!(var_names, varname)
            push!(data, Array[single_lat_lon[varname]][1][:])
        end

        return reduce(hcat, data)'
    end
end

方法2：绕过宏，直接用索引定位 #

完全避开宏调用，手动通过维度索引定位目标经纬度：

function data_for_lat_lon(ds, region, lat_lon_pair)
    println("have latlon piar ", lat_lon_pair)
    lat = lat_lon_pair[1]
    lon = lat_lon_pair[2]
    data = []
    if(ArchGDAL.contains(region[1], ArchGDAL.createpoint(lon, lat)))
        println(lat, " ", lon)
        # 直接查找经纬度对应的索引
        lat_idx = findfirst(==(lat), ds["latitude"][:])
        lon_idx = findfirst(==(lon), ds["longitude"][:])
        # 手动提取对应位置的数据
        for (varname, var) in ds
            if varname in ["latitude", "longitude", "time"]
                continue
            end
            var_data = var[lon_idx, lat_idx, :]
            push!(var_names, varname)
            push!(data, var_data[:])
        end

        return reduce(hcat, data)'
    end
end

说明 #

• 方法1通过@eval打破宏的静态展开限制，让变量值直接嵌入宏生成的代码，解决了宏卫生性带来的作用域问题。
• 方法2直接操作数据集索引，完全避开宏的编译时机问题，更适合函数内的动态参数场景。

内容的提问来源于stack exchange，提问作者Scott Chamberlin