在这个例子中如何避免代码重复?

时间:2011-10-19 04:16:51

标签: c++

我有两个班级,BaseDerived。我提供了一个函数,它将函数应用于Base的所有元素,另一个函数将函数应用于Derived的所有元素:

#include <iostream>
#include <vector>

class Base {
  typedef std::vector<int> Vec;
  typedef Vec::const_iterator CIter;
  CIter begin() const;
  CIter end() const;
};

class Derived : public Base {
  void add(int);
};


template<class F>
void processBase(const Base& b, F& f) {
  for (Base::CIter it = b.begin(); it != b.end(); ++it) {
    f(b, *it);
  }
}

template<class F>
void processDerived(Derived* d, F& f) {
  for (Base::CIter it = d->begin(); it != d->end(); ++it) {
    f(d, *it);
  }
}

如何让one函数调用另一个函数,以便我没有代码重复? (我只给出了一个最小的例子;在我的实际代码中,for循环有点复杂,并且在两个函数中完全重复了该代码。)

我是否必须模拟processBase功能?或者有没有办法使用铸造?

1 个答案:

答案 0 :(得分:5)

假设const Base&Derived*的区别是错误,是的,您需要一个函数模板来调用具有正确参数类型的仿函数:

template <typename Collection, typename F>
void processCollection (Collection& col, F& f) {
    for (typename Collection::CIter i=col.begin(); i!=col.end(); ++i) {
        f (col, *i);
    }
}

processCollection适用于const或非constBaseDerived个对象。