zoukankan      html  css  js  c++  java
  • C++雾中风景16:std::make_index_sequence, 来试一试新的黑魔法吧

    C++14在标准库里添加了一个很有意思的元函数: std::integer_sequence。并且通过它衍生出了一系列的帮助模板:
    std::make_integer_sequence, std::make_index_sequence, std:: index_sequence_for。在新的黑魔法的加持下,它可以帮助我们完成在编译期间获取了一组编译期整数的工作。
    接下来请系好安全带,准备发车,和大家聊聊新的黑魔法:std::make_index_sequence

    1.what's std::make_index_sequence

    1.1 举个栗子

    笔者这里先从一个简单的例子展开,先带大家看看std::make_index_sequence是如何使用的。

    在C++之中有一个很常见的需求,定义一组编译期间的数组作为常量,并在运行时或者编译时利用这些常量进行计算。现在假如我们需编译期的一组1到4的平方值。你会怎么办呢?

    嗯.... 思考一下,可以这些写:

    constexpr static size_t const_nums[] = {0, 1, 4, 9, 16};
    
    int main() {
        static_assert(const_nums[3] == 9); 
    }
    

    Bingo, 这个代码肯定是正确的,但是如果4扩展到了20或者100?怎么办呢?

    嗯~~,先别着急骂脏话,我们可以用std::make_index_sequencestd::index_sequence来帮助我们实现这个逻辑:

    template <size_t ...N>
    static constexpr auto square_nums(size_t index, std::index_sequence<N...>) {
        constexpr auto nums = std::array{N * N ...};
        return nums[index];
    }
    
    template <size_t N>
    constexpr static auto const_nums(size_t index) {
        return square_nums(index, std::make_index_sequence<N>{});
    }
    
    int main() {
        static_assert(const_nums<101>(100) == 100 * 100); 
    }
    

    这代码咋看之下有些吓人,不过没关系,我们来一点点的庖丁解牛的解释清楚它是如何工作的。

    1.2 做个解释

    我们来拆解一下1.1的代码,首先我们定义了一个constexpr的静态函数const_nums。它通过我们本文的主角std::make_index_sequence来构造了一组0,1,2,3 .... N - 1的一组编译器的可变长度的整数列。(注意,这里调用std::make_index_sequence{}的构造函数没有任何意义,纯粹只是利用了它能够生成编译期整数列的能力。)
    const_nums函数

    接着我们来看squere_num函数,这就是我们实际进行平方计算,并生成编译期静态数组的地方了,它的实现很简单,就是依次展开通过std::make_index_sequence生成的数字,并进行平方计算,最后塞到std::array的构造函数之中进行构造。

    2. How std::make_index_sequence

    通过一个简单的栗子,大家想必已经见识到这个新的黑魔法的独特之处了。接下来,我们进一步的来剖析它的实现吧。

    2.1 std::integer_sequence
    template< class T, T... Ints >
    class integer_sequence;
    

    要了解std::make_index_sequence是如何工作的,就得先看看它的基础类:std::integer_sequence。由上面的代码看,它很简单,就是一个int类型,加上一组int数字,其实原理就是生成一组T类型的编译期间数字序列。它本质上就是个空类,我们就是要获取这个编译期的数字序列。

    2.2 std::index_sequence
    template<size_t.. Ints>
    using index_sequence = std::integer_sequence<size_t,Ints...>;
    

    通常我们不会直接使用std::integer_sequence,而是通过定义一个size_tstd::integer_sequnece命名为index_sequence

    2.3 std::make_index_sequence

    这里就是生成了一组数字序列0,1,2,3...N - 1的一组std::index_sequence。然后就可以利用这组序列的数字做任何我们想做的事情了。

    那么问题来了,std::make_index_sequence是如何实现的呢?

    • 可以通过元编程,生成N个元函数类,依次生成0到N - 1的序列,感兴趣的话可以参考这个链接
    • 实际是由编译器内部在编译期间实现的,并不是基于现有的元编程的逻辑。

    3.std::make_index_sequence与std::tuple

    通过第二节的介绍,想必大家应该了解了std::make_index_sequence的实现原理了。接下来将介绍它最为重要的使用场景:与tuple的结合。

    现在请大家思考一个问题:如何遍历一个std::tuple。(不能使用C++17的std::apply

    这个时候就要再次请出我们今天的主角,使用std::make_index_sequnce和lambda表达式来完成这个工作了。我们来看下面这部分代码:

    template <typename Tuple, typename Func, size_t ... N>
    void func_call_tuple(const Tuple& t, Func&& func, std::index_sequence<N...>) {
        static_cast<void>(std::initializer_list<int>{(func(std::get<N>(t)), 0)...});
    }
    
    template <typename ... Args, typename Func>
    void travel_tuple(const std::tuple<Args...>& t, Func&& func) {
        func_call_tuple(t, std::forward<Func>(func), std::make_index_sequence<sizeof...(Args)>{});
    }
    
    int main() {
        auto t = std::make_tuple(1, 4.56, "happen lee");
        travel_tuple(t, [](auto&& item) {
            std::cout << item << ",";
        });
    }
    
    • 这个代码首先定义了一个travel_tuple的函数,并且利用了std::make_index_sequence将tuple类型的参数个数进行了展开,生成了0到N - 1的编译期数字。

    • 接下来我们再利用func_call_tuple函数和展开的编译期数字,依次调用std::get<N>(tuple),并且通过lambda表达式依次的调用,完成了遍历tuple的逻辑。

    嗯,标准库表示它也是这样想的,所以C++17利用了std::make_index_sequence实现了std::apply,开启了满屏幕堆满tuple的C++新时代了~~

    4.小结

    C++14新提供的std::make_index_sequence给了我们在编译期操作tuple提供了更加便利的工具,并且在编译期间的整数列也能够帮助我们实现更多新的黑魔法。

    大家可以尝试自己用元编程实现了一个std::make_index_sequence, 笔者觉得这是一个很有意思的挑战。

    关于std::make_index_sequence就聊到这里。希望大家能够有所收获,笔者水平有限。成文之处难免有理解谬误之处,欢迎大家多多讨论,指教。

    5.参考资料

    cppreference
    make_index_sequence的原理

  • 相关阅读:
    ssh-keygen的使用方法(无密码访问)
    ubuntu solute two different terminals cmd
    ubuntu 查看系统是32位还是64位
    pyplot 绘图与可视化
    python 正则表达式的处理
    python&pandas 与mysql 连接
    Ubuntu 11.10 H3C iNode 客户端安装
    Vijos1055(极大子矩阵)
    Vijos1055(极大子矩阵)
    luoguP2701 [USACO5.3]巨大的牛棚Big Barn(极大子矩阵)
  • 原文地址:https://www.cnblogs.com/happenlee/p/14219925.html
Copyright © 2011-2022 走看看