在编译时计算一组常量表达式的最大值

问题描述 投票:4回答:1

我试图在Rust程序宏(派生宏)内的编译时计算一组常量的最大值。

宏看起来像:

fn get_max_len() -> TokenStream {
    // Each TokenStream represents a constant expression
    let len: Vec<TokenStream> = get_constant_lengths();

    quote! {
      // #(#len),* gets expanded to #len[0], #len[1], #len[2]...
      const LEN: usize = std::cmp::max(#(#len),*);
    }
}

问题是std::cmp::max是一个函数,因此不能在常量表达式中使用(至少在const fn稳定之前 - 我想尽可能保持稳定的Rust)。

如何在编译时计算一组常量的最大值?

我或许可以编写一个max!宏,基本上可以递归地构建一个巨大的ifs链,但我希望那里有一个更清洁的解决方案。

rust rust-macros
1个回答
5
投票

虽然常量评估不支持if或其他控制流,但有一种方法可以根据二进制条件选择值:

[a, b][(a < b) as usize]

这是做什么的

  • 创建要在其中选择的两个元素的数组
  • 创建一个任意的布尔表达式
  • 把悲伤的表情投向了qazxsw poi
  • 使用该值索引到上面创建的数组中

如果条件是usize,则选择第一个元素,如果条件是false,则选择第二个元素。

虽然这个方案理论上可以通过在多个铸造的trues上通过数学运算计算索引来扩展到任意长度的数组,但只是去功能方式并嵌套上面的表达式似乎更简单:

bool

从Rust 1.31开始,const fn max(a: usize, b: usize) -> usize { [a, b][(a < b) as usize] } const MAX: usize = max(max(max(5, 6), 42), 3); 可用于稳定的编译器。

© www.soinside.com 2019 - 2024. All rights reserved.