Nim中如何遍历多个元组实现算术运算并限制元组元素类型

1. 基础遍历元组的运算符实现 #

直接用Nim的静态遍历特性即可实现对任意长度同类型元组的逐元素运算，不需要手动逐位编写代码，完整实现如下：

import std/typetraits

# 通用逐元素加法实现
proc `+`[T: tuple](a: T, b: T): T =
  # 静态遍历元组所有下标，编译时自动展开为对应位置的运算代码
  for i in static 0..<T.len:
    result[i] = a[i] + b[i]

# 测试代码
echo (1, 1) + (2, 2)         # 输出 (3, 3)
echo (1.2, 3.4, 5.6) + (2.3, 4.5, 6.7) # 输出 (3.5, 7.9, 12.3)

2. 批量生成多运算符实现 #

如果需要同时实现-、*等多个运算符，可以用模板批量生成，避免重复代码：

import std/typetraits

# 批量生成元组运算符的模板
template genTupleArithmeticOp(op: untyped): untyped =
  proc op[T: tuple](a: T, b: T): T =
    for i in static 0..<T.len:
      result[i] = op(a[i], b[i])

# 一键生成所需的运算符
genTupleArithmeticOp(`+`)
genTupleArithmeticOp(`-`)
genTupleArithmeticOp(`*`)

# 测试
echo (3, 5) - (1, 2) # 输出 (2, 3)
echo (2, 3, 4) * (3, 2, 1) # 输出 (6, 6, 4)

3. 泛型约束实现 #

3.1 限制元组所有元素为float #

通过静态类型谓词实现约束：

import std/typetraits

# 定义「所有元素都是float的元组」类型约束
type AllFloatTuple[T] = T is tuple and allIt(T.TupleFields, it is float)

# 仅接受全float元组的加法实现
proc `+`[T: tuple; AllFloatTuple[T]](a: T, b: T): T =
  for i in static 0..<T.len:
    result[i] = a[i] + b[i]

echo (1.1, 2.2) + (3.3, 4.4) # 正常运行
# echo (1, 2) + (3, 4) # 编译报错，不满足全float约束

3.2 限制元组所有元素为int或float #

修改类型约束即可：

import std/typetraits

# 定义「所有元素都是int或float的元组」类型约束
type AllNumberTuple[T] = T is tuple and allIt(T.TupleFields, it is int or it is float)

# 仅接受数值元素元组的加法实现
proc `+`[T: tuple; AllNumberTuple[T]](a: T, b: T): T =
  for i in static 0..<T.len:
    result[i] = a[i] + b[i]

echo (1, 2) + (3, 4) # 正常运行
echo (1.1, 2.2) + (3.3, 4.4) # 正常运行
# echo ("a", "b") + ("c", "d") # 编译报错，元素不是数值类型

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