Nim中的条件类型：如何从整数函数参数获取半长无符号整数类型？

问题需求 #

需要实现类似C++中std::conditional_t的条件类型逻辑：根据输入整数类型的字节大小，返回其一半长度的无符号整数类型（例如8字节整数返回uint32，4字节返回uint16，2字节返回uint8）。

用户提供的代码 #

无法运行的Nim示例代码 #

macro GetHalfUInt(anyInt : untyped ): untyped =
  when sizeof(anyInt) == 8:
    uint32
  else when sizeof(anyInt) == 4:
    uint16
  else when sizeof(anyInt) == 2:
    uint8


proc getHighBitsAsHalfUInt[AnyInt](x: AnyInt) : GetHalfUInt(AnyInt) =
  result = (x shr (sizeof(AnyInt) * 4)).GetHalfUInt(AnyInt)

可正常工作的C++参考代码 #

template<class AnyInt>
struct GetHalfUInt {
    static_assert( std::is_integral_v<AnyInt>, "Must be Int type!");
    using type = std::conditional_t<sizeof(AnyInt) == 8, uint32_t, 
                    std::conditional_t<sizeof(AnyInt) == 4, uint16_t, uint8_t> >;
};

template<class AnyInt>
auto getHighBitsAsHalfUInt(AnyInt x) {
    using Res = typename GetHalfUInt<AnyInt>::type;
    return Res(x >> (sizeof(AnyInt)*4));
}

解决方案 #

Nim的泛型和编译期元编程模型与C++不同，无需用macro实现，直接利用编译期类型函数或泛型proc内的when分支即可完成需求：

方案1：使用编译期类型函数（推荐） #

先定义一个类型函数，根据输入类型的大小映射到目标无符号类型，同时添加编译期断言确保输入为整数类型：

type
  GetHalfUInt*[T] = type
    when sizeof(T) == 8: uint32
    elif sizeof(T) == 4: uint16
    elif sizeof(T) == 2: uint8
    else:
      static: error("Unsupported integer type size (only 2/4/8-byte integers are allowed)")

proc getHighBitsAsHalfUInt*[T](x: T): GetHalfUInt[T] =
  static:
    when not isInteger(T):
      error("Parameter must be an integer type")
  # 直接用类型构造完成转换，右移位数为类型字节数*4（即一半的比特数）
  result = GetHalfUInt[T](x shr (sizeof(T) * 4))

方案2：直接在泛型proc内处理 #

如果不需要单独的类型函数，也可以直接在泛型proc中用when分支指定返回类型和逻辑：

proc getHighBitsAsHalfUInt*[T](x: T): auto =
  static:
    when not isInteger(T):
      error("Parameter must be an integer type")
    when sizeof(T) not in {2,4,8}:
      error("Unsupported integer type size (only 2/4/8-byte integers are allowed)")
  
  when sizeof(T) == 8:
    result = uint32(x shr 32)
  elif sizeof(T) == 4:
    result = uint16(x shr 16)
  elif sizeof(T) == 2:
    result = uint8(x shr 8)

原代码问题说明 #

1. Macro使用错误：原macro试图直接返回类型名，但Nim的macro需要返回语法树节点，且泛型类型参数AnyInt无法直接作为macro的参数处理（macro处理的是语法节点而非类型）。
2. 类型转换写法错误：Nim中类型转换需用Type(value)或cast[Type](value)，而非.GetHalfUInt(AnyInt)这种语法。
3. 缺少编译期检查：原代码未验证输入是否为整数类型，也未处理不支持的类型大小情况。

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