检查容器类型

Check for container type

本文关键字：类型 检查      更新时间：2023-10-16

如何检查容器类型?例如

template <class Container1>
void func (Container1 &cont_1)
{
  if (cont_1 is list container)
    //do stuff
  if (cont_1 is vector container)    
    //do other stuff
}

我仅有的两种可能是list和vector。注意，我不知道list/vector(即vector<char>或vector<int>等)中值的类型，因此我只想获得获取向量或列表的信息。我遇到了typeid和type info，但没有真正完成。

可以使用函数重载来实现:

template<typename T>
void func(std::vector<T>& vec)
{
    //Do something with vector
}
template<typename T>
void func(std::list<T>& list)
{
    //Do something with list
}

或者使用typeid，这可能不太理想，因为无论哪种情况下的代码都必须对std::vector和std::list都可编译，因为模板在编译时是已知的，即使分支可能不会在std::list上执行，编译器在那时也不知道这一点，因此它将编译失败，试图在std::list上应用std::vector操作。

template<template<typename, typename> class C, typename T, typename Alloc>
void func(C<T, Alloc>& container)
{
    if (typeid(container).hash_code() == typeid(std::vector<T, Alloc>&).hash_code())
        ; //Do something with vector
    else if (typeid(container).hash_code() == typeid(std::list<T, Alloc>&).hash_code())
        ; //Do something with list
}

使用重载的方法当然更好，但出于完整性考虑，还有另一种使用sfinae机制的方法。在c++11之前，示例代码是这样的:

#include <iostream>
#include <vector>
#include <list>
template <bool V, class T = void>
struct enable_if { };
template <class T>
struct enable_if<true, T> {
   typedef T type;
};
struct true_type {
   static const bool value = true;
};
struct false_type {
   static const bool value = false;
};
template <class T>
struct is_list: false_type {};
template <class T>
struct is_list<typename std::list<T> >: true_type { };
template <class T>
struct is_vector: false_type { };
template <class T>
struct is_vector<typename std::vector<T> >: true_type { };
template <class C>
typename enable_if<is_vector<C>::value>::type func(C &c) {
   // c is a vector here
   std::cout << "This is a vector" << std::endl;
}
template <class C>
typename enable_if<is_list<C>::value>::type func(C &c) {
   // c is a list here
   std::cout << "This is a list" << std::endl;
}
int main() {
   std::vector<int> v;
   func(v);
   std::list<int> l;
   func(l);
}

在c++11中会变得简单一些:

#include <iostream>
#include <vector>
#include <list>
#include <type_traits>
template <class T>
struct is_list: std::false_type {};
template <class T>
struct is_list<typename std::list<T>>: std::true_type { };
template <class T>
struct is_vector: std::false_type { };
template <class T>
struct is_vector<typename std::vector<T>>: std::true_type { };
template <class C>
typename std::enable_if<is_vector<C>::value>::type func(C &c) {
   // c is a vector here
   std::cout << "This is a vector" << std::endl;
}
template <class C>
typename std::enable_if<is_list<C>::value>::type func(C &c) {
   // c is a list here
   std::cout << "This is a list" << std::endl;
}
int main() {
   std::vector<int> v;
   func(v);
   std::list<int> l;
   func(l);
}

您可以使用可变模板编写一些测试:

#include <vector>
// is_vector
// =========
namespace Detail {
    template<typename T>
    struct is_vector_type : std::false_type
    {};
    template<typename ... Types>
    struct is_vector_type<std::vector<Types...>>  : std::true_type
    {};
}
template<typename T>
struct is_vector_type : Detail::is_vector_type<typename std::decay<T>::type>
{};
template<typename T>
constexpr bool is_vector(T&&) {
    return is_vector_type<T>();
}
// Test
// ====
#include <iostream>
template <class Container>
void func (Container &container)
{
  if (is_vector_type<Container>()) {
    std::cout << "vectorn";
  }
  // or
  if (is_vector(container)) {
    std::cout << "vectorn";
  }
  // ...
}
int main() {
    std::vector<int> v;
    func(v);
    static_assert(is_vector(v), "");
}
相关文章：