采用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);
}