STL容器的枚举列表

Enum list to STL container

本文关键字：枚举 列表 STL      更新时间：2023-10-16

有人能改进我的这个功能吗？enumToSTLContainer用于返回元素来自指定枚举列表的任何STL容器。

#include <iostream>
#include <vector>
#include <set>
enum MyEnum {A, B, C, D, E, FirstMyEnum = A, LastMyEnum = E};
template <typename CONTAINER>
CONTAINER enumToSTLContainer (typename CONTAINER::value_type first, typename CONTAINER::value_type last) {
    CONTAINER v;
    for (typename std::underlying_type<typename CONTAINER::value_type>::type N = first;  N <= last;  N++)
        v.push_back(static_cast<typename CONTAINER::value_type>(N));  // But only works for STL containers with push_back defined.
    return v;
}
int main() {
    const std::vector<MyEnum> enumVector = enumToSTLContainer<std::vector<MyEnum>> (FirstMyEnum, LastMyEnum);
    for (MyEnum x : enumVector)
        std::cout << x << ' ';  // 0 1 2 3 4
//  const std::set<MyEnum> enumSet = enumToSTLContainer<std::set<MyEnum>> (FirstMyEnum, LastMyEnum);  // won't compile
}

上述问题。仅适用于定义了push_back的STL容器，需要指定枚举列表的第一个和最后一个元素，如果枚举值不连续则不起作用。。。

也许可以使用可变模板获得initializer_list，然后返回所需的STL容器？使用std:：integer_sequence从枚举中生成元组，或者在枚举元素不连续的情况下使自己的integer_sequet与它们相对应？

好的，我用以下内容得到了所需的属性，但它使用了枚举基础值从零开始并且是连续的假设。解决方案是用C++11而不是C++14，所以这里的每个人都应该能够编译它

#include <iostream>
#include <vector>
#include <set>
enum MyEnum {A, B, C, D, E, NumElements};
template <std::size_t...> struct index_sequence {};
template <std::size_t N, std::size_t... Is>
struct make_index_sequence_helper : make_index_sequence_helper<N-1, N-1, Is...> {};
template <std::size_t... Is>
struct make_index_sequence_helper<0, Is...> {
    using type = index_sequence<Is...>;
};
template <std::size_t N>
using make_index_sequence = typename make_index_sequence_helper<N>::type; 
template <typename RETURN_TYPE, std::size_t... TYPES>
inline RETURN_TYPE helper (index_sequence<TYPES...>) {
    return {static_cast<typename RETURN_TYPE::value_type>(TYPES)...};
}
template <typename CONTAINER, std::size_t N>
CONTAINER enumToSTLContainer() {
    return helper<CONTAINER> (make_index_sequence<N>());
}
int main() {
    const std::vector<MyEnum> enumVector = enumToSTLContainer<std::vector<MyEnum>, NumElements>();
    for (MyEnum x : enumVector)
        std::cout << x << ' ';  // 0 1 2 3 4
    const std::set<MyEnum> enumSet = enumToSTLContainer<std::set<MyEnum>, NumElements>();
    std::cout << "nn";
    for (MyEnum x : enumSet)
        std::cout << x << ' ';  // 0 1 2 3 4
}

因此，使用第二个解决方案，我们不再局限于具有push_back的STL容器。如果有人能向我展示如何改进枚举基础值不连续的解决方案，我将不胜感激。理想情况下也不需要NumElements。我认为只有make_index_sequence_helper需要根据MyEnum基础值进行修改？

我唯一能想到的概括就是糟糕的：

template <typename CONTAINER, std::size_t... Is>
CONTAINER enumNonContiguousToSTLContainer() {
    return helper<CONTAINER> (index_sequence<Is...>());
}
相关文章：