STL中仿函数(functors)、类成员和mem_fun的使用

仿函数（functor）: 重载了"()"运算符的struct，例如：

struct D {
  D(int i=0){num=i;}
  int num;
};
struct print_D{
  void operator()(const D* d)const{
      cout<<"I am D. my num="<<d->num<<endl;
    }
};

int main()
{
  vector<D*> V;

  V.push_back(new D(1));
  V.push_back(new D(2));
  V.push_back(new D);
  V.push_back(new D(3));

  for_each(V.begin(), V.end(), print_D());
}
编译输出：

I am D. my num=1
I am D. my num=2
I am D. my num=0
I am D. my num=3

 

如果使用mem_fun，会方便很多：

struct D {
  D(int i=0){num=i;}
  void print() { cout << "I'm a D. my num=" << num<< endl; }
  int num;
};

int main()
{
  vector<D*> V;

  V.push_back(new D(1));
  V.push_back(new D(2));
  V.push_back(new D);
  V.push_back(new D(3));

  for_each(V.begin(), V.end(), mem_fun(&D::print));
}

那看看源码是怎么回事，在SGI STL的stl_function.h:

template <class _Ret, class _Tp>
inline mem_fun_t<_Ret,_Tp> mem_fun(_Ret (_Tp::*__f)())
{ return mem_fun_t<_Ret,_Tp>(__f); }

原来mem_fun返回的是一个对象:mem_fun_t<_Ret,_Tp>.(不要嫌人家命名太怪异).那mem_fun_t<_Ret,_Tp>又是什么东东？还是看源码：

template <class _Ret, class _Tp>
class mem_fun_t : public unary_function<_Tp*,_Ret> {
public:
    explicit mem_fun_t(_Ret (_Tp::*__pf)()) : _M_f(__pf) {}
    _Ret operator()(_Tp* __p) const { return (__p->*_M_f)(); }
private:
    _Ret (_Tp::*_M_f)();
};

原来mem_fun_t就是一个functor，这下就满足了for_each的要求了。其调用流程是这样的，for_each把vector中的元素传送给mem_fun，mem_fun自己产生一个仿函数mem_fun_t，然后仿函数调用其重载的()。过程就这么简单。当然你不能对其他类的成员函数进行绑定，因为在for_each调用过程中，会传递其*iterator值，如果是其他类的成员函数，那么这个类的对象无法传入，当然就无法完成任务了。

这里使用的是vector<D*> V; 在mem_fun_t构造函数中，刚好需要指针，如果不是D*, 而是使用vector<D> V; 还能用吗？

这是你需要使用的是mem_fun_ref。把程序改成：

struct D {
  D(int i=0){num=i;}
  void print() { cout << "I'm a D. my num=" << num<< endl; }
  int num;
};

int main()
{
  vector<D> V;

  V.push_back(D(1));
  V.push_back( D(2));
  V.push_back( D());
  V.push_back( D(3));

  for_each(V.begin(), V.end(), mem_fun_ref(&D::print));
}

mem_fun对于一些多态的虚函数也十分有用，注意看下面的例子：

struct B {
  virtual void print() = 0;
};

struct D1 : public B {
  void print() { cout << "I'm a D1" << endl; }
};

struct D2 : public B {
  void print() { cout << "I'm a D2" << endl; }
};

int main()
{
  vector<B*> V;

  V.push_back(new D1);
  V.push_back(new D2);
  V.push_back(new D2);
  V.push_back(new D1);

  for_each(V.begin(), V.end(), mem_fun(&B::print));
}

mem_fun1和mem_fun1_ref，他们可以使用绑定一个参数的类成员。

 

Code

 

posted @ 2009-06-21 15:56  辛勤耕耘  阅读(565)  评论(0)  编辑  收藏  举报