c++ - 某种生成循环

我有这段代码：

template<class T, class color_type>
void assign_channels(T* ptr, const color_type& color) {
    *(ptr + 0) = get_channel<0>(color);

    if constexpr(1 < color_type::size) {
        *(ptr + 1) = get_channel<1>(color);
    }
    if constexpr(2 < color_type::size) {
        *(ptr + 2) = get_channel<2>(color);
    }
    if constexpr(3 < color_type::size) {
        *(ptr + 3) = get_channel<3>(color);
    }
    if constexpr(4 < color_type::size) {
        *(ptr + 4) = get_channel<4>(color);
    }
    if constexpr(5 < color_type::size) {
        *(ptr + 5) = get_channel<5>(color);
    }
    if constexpr(6 < color_type::size) {
        *(ptr + 6) = get_channel<6>(color);
    }
    if constexpr(7 < color_type::size) {
        *(ptr + 7) = get_channel<7>(color);
    }
}

它有两个明显的缺陷：


除非我的频道不超过8个，否则它将无法正常工作；
主要是样板。


C ++中是否可以通过某种循环来重写它？我认为可以使用重复的模板化结构来完成这项工作，但是它并不是很可读。我该怎么办？

最佳答案

在C ++ 17中使用折叠表达式和std::index_sequence

template<class T, class color_type, size_t... Is>
void assign_channels(T* ptr, const color_type& color, std::index_sequence<Is...>) {
    ((ptr[Is] = get_channel<Is>(color)), ...);
}

template<class T, class color_type>
void assign_channels(T* ptr, const color_type& color) {
    assign_channels(ptr, color, std::make_index_sequence<color_type::size>{});
}

在C ++ 17之前，您必须将fold表达式重写为递归。

关于c++ - 某种生成循环，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/47240147/