c++ - 模板函数采用 std::vector 或 std::array

Question

我有一个函数，目前接受 2 个可以包含任何普通旧数据的向量......

template <class T>
void addData(const vector<T>& yData, vector<T> xData)
{ .. }

问题：

• 考虑到这些容器采用不同数量的模板参数，是否可以将其修改为采用两个std::array或两个std::vector，甚至两者的组合？

Answer 1

当然，这只是创建合适的类型特征的问题。该示例仅使用f()带有一个参数的函数，但扩展为采用任意数量的参数是微不足道的。

#include <array>
#include <vector>
#include <deque>
#include <utility>
#include <cstddef>

template <typename T>
struct is_array_or_vector {
    enum { value = false };
};

template <typename T, typename A>
struct is_array_or_vector<std::vector<T, A>> {
    enum { value = true };
};

template <typename T, std::size_t N>
struct is_array_or_vector<std::array<T, N>> {
    enum { value = true };
};

template <typename T>
typename std::enable_if<is_array_or_vector<T>::value>::type
f(T const&)
{
}

int main()
{
    f(std::vector<int>());    // OK
    f(std::array<int, 17>()); // OK
    f(std::deque<int>());     // ERROR
}

Answer 2

为什么不直接使用它，它适用于任何使用随机访问迭代器的容器，包括普通的旧数组。如果您可以使用迭代而不是索引，那么您也可以取消随机访问要求。

template <typename Cnt1, typename Cnt2>
void addData(const Cnt1& yData, Cnt2 xData) // is pass-by-value intended?
{
    using std::begin;
    using std::end;
    typedef decltype(*begin(yData)) T;
    const auto sizeY = end(yData) - begin(yData);
    const auto sizeX = end(xData) - begin(xData);
    // ...
}

C++03 版本（不支持普通旧数组）：

template <typename Cnt1, typename Cnt2>
void addData(const Cnt1& yData, Cnt2 xData) // is pass-by-value intended?
{
    typedef Cnt1::value_type T;
    const size_t sizeY = yData.end() - yData.begin();
    const size_t sizeX = xData.end() - xData.begin();
    // ...
}

Answer 3

另一种解决方案：

#include <iostream>
#include <vector>
#include <array>

using std::vector;
using std::array;

template <typename Container>
struct container_helper; // undefined

template <typename T>
struct container_helper<vector<T>>
{    
    explicit container_helper(vector<T>& data)
    : _data(data)
    {}

    T* get_data()
    { return &_data[0]; }

    size_t get_size()
    { return _data.size(); }

private:
    vector<T>& _data;
};

template <typename T, size_t N>
struct container_helper<array<T,N>>
{
    explicit container_helper(array<T,N>& data)
    : _data(data)
    {}

    T* get_data()
    { return &_data[0]; }

    size_t get_size()
    { return N; }

private:
    array<T,N>& _data;
};


template <typename Container1, typename Container2>
void add_data(Container1& c1, Container2& c2)
{
    container_helper<Container1> c1_helper(c1);
    container_helper<Container2> c2_helper(c2);

    /* do whatever you want with the containers */
    std::cout << "c1 size " << c1_helper.get_size() << std::endl;
    std::cout << "c2 size " << c2_helper.get_size() << std::endl;
}


int main()
{
    vector<int > v_ints(3);
    array<int, 2> a_ints;

    add_data(v_ints, a_ints);
}