我正在尝试创建一个类,它将从一组向量中绘制元素(并且还将这些向量作为容器放在类中),但我觉得在管理具有很多函数的向量时,例如vectorOneAdd,vectorTwoAdd使用为了向向量添加元素是没有意义的。必须有一个更好的方法,这就是为什么我在这里问,我听说你可以使用模板来做,但我不太清楚如何。需要帮助。不希望有很多无意义的代码。
我的意思如下:
class Cookie
{
std::vector<Chocolate> chocolateContainer;
std::vector<Sugar> sugarContainer;
void chocolateVectorAdd(Chocolate element); // first function adding to one vector
void sugarVectorAdd(Sugar element); // second function adding to another vector
}
请使用示例代码,谢谢:)
答案 0 :(得分:6)
有很多函数,比如vectorOneAdd,vectorTwoAdd用于向向量添加元素是没有意义的。必须有更好的方法
有:
class Cookie {
std::vector<Chocolate> chocolateContainer;
std::vector<Sugar> sugarContainer;
private:
template<typename T>
std::vector<T>& get_vector(const T&); // not implemented but particularized
// write one of these for each vector:
template<>
std::vector<Chocolate>& get_vector(const Chocolate&) { return chocolateVector; }
template<>
std::vector<Sugar>& get_vector(const Sugar&) { return sugarVector; }
public:
template<typename T>
void add(T element) {
auto& v = get_vector(element);
v.push_back(std::move(element));
}
};
答案 1 :(得分:2)
以下将解决您的问题(需要C ++ 11)
#if 1 // std::get<T>(tuple) is not in C++11
// Helper to retrieve index to be able to call std::get<N>(tuple)
template <typename T, typename ... Ts> struct get_index;
template <typename T, typename ... Ts>
struct get_index<T, T, Ts...> : std::integral_constant<std::size_t, 0> {};
template <typename T, typename Tail, typename ... Ts>
struct get_index<T, Tail, Ts...> :
std::integral_constant<std::size_t, 1 + get_index<T, Ts...>::value> {};
#endif
template <typename ... Ts>
class Data
{
public:
template <typename T>
void push_back(const T& x) { return getVector<T>().push_back(x); }
template <typename T>
std::size_t size() const { return getVector<T>().size(); }
private:
template <typename T>
const std::vector<T>& getVector() const { return std::get<get_index<T, Ts...>::value>(items); }
template <typename T>
std::vector<T>& getVector() { return std::get<get_index<T, Ts...>::value>(items); }
private:
std::tuple<std::vector<Ts>...> items;
};
测试它:
class A{};
class B{};
int main()
{
Data<A, B> d;
d.push_back(A());
d.push_back(B());
return 0;
}