如何在Julia中高效通过索引访问函数并优化现有实现？

我有一组用于处理数据的函数，目前将它们存储在全局数组中以通过索引访问，代码如下：

const global _rot01(x)::Tuple{Int64, Int64, Int64} = ( x[1],  x[2],  x[3])
const global _rot02(x)::Tuple{Int64, Int64, Int64} = ( x[2],  x[3],  x[1])
...
const global _rot24(x)::Tuple{Int64, Int64, Int64} = (-x[1], -x[3], -x[2])

const global Rotations = [
    _rot01,
    ...
    _rot24
]

但我希望对此进行优化，因为听说应避免使用全局变量。switch语句可高效选择对应函数，但Julia中没有该语句。请问函数向量是唯一实现方式吗？这种方式效率如何？另外，我是否应将现有函数替换为正式的函数定义？

函数向量不是唯一实现方式 #

除了函数向量，还有两种更贴合Julia特性的替代方案：

• 枚举+多重分派：定义枚举类型标记旋转类型，再通过分派机制绑定对应逻辑。编译时就能确定调用的方法，完全规避全局变量问题，性能拉满：

  @enum RotationID rot01 rot02 ... rot24
  
  function apply_rotation(id::RotationID, x)
      return apply_rotation(Val(id), x)
  end
  
  apply_rotation(::Val{rot01}, x) = (x[1], x[2], x[3])
  apply_rotation(::Val{rot02}, x) = (x[2], x[3], x[1])
  ...
  apply_rotation(::Val{rot24}, x) = (-x[1], -x[3], -x[2])
  

• 生成函数：如果旋转规则有统一数学规律，用生成函数动态生成计算逻辑，无需手动定义24个函数，代码更简洁，效率也有保障：

  @generated function apply_rotation(n::Int, x)
      # 根据n的数值生成对应的旋转表达式
      if n == 1
          return :( (x[1], x[2], x[3]) )
      elseif n == 2
          return :( (x[2], x[3], x[1]) )
      ...
      elseif n ==24
          return :( (-x[1], -x[3], -x[2]) )
      end
  end
  

函数向量的效率表现 #

你当前写法中，Rotations已经声明为const，Julia会对这种类型确定的函数数组做充分优化——索引访问后调用函数的效率和直接调用单个函数几乎无差别。但如果去掉const，全局数组的类型不确定性会导致性能下降，所以保持const是关键。

是否替换为正式函数定义 #

性能上，你现在用的短形式函数定义和正式function关键字定义没有区别，但正式定义在代码可读性、可维护性上更有优势，尤其是后续需要扩展函数逻辑时。另外更推荐的做法是把这些函数封装到模块里，而非暴露在全局作用域：

module RotationUtils
    export Rotations

    function rot01(x)::Tuple{Int64, Int64, Int64}
        return (x[1], x[2], x[3])
    end

    function rot02(x)::Tuple{Int64, Int64, Int64}
        return (x[2], x[3], x[1])
    end

    ...

    function rot24(x)::Tuple{Int64, Int64, Int64}
        return (-x[1], -x[3], -x[2])
    end

    const Rotations = [rot01, rot02, ..., rot24]
end

模块封装能把全局变量的影响限制在模块内部，更符合Julia的最佳实践。

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