C++之流运算符重载

之前说到流的时候提到过，cin和cout是一种流对象，对于这样的一句代码：

cout << a << b << endl;

实际的步骤是这样的，a先流到cout里面，然后b再流到cout的里面，最后endl流到cout里面，可以看看出来“<<”是一个双目操作符，那我们是不是也可以用友元重载和成员函数重载来分别实现呢？

这里要说一下，”<<”和“>>”都只能使用友元重载实现，不能用成员函数重载，稍微解释一下为什么：

“<<”和”>>”的左边分别是cout和cin，如果想用成员函数重载实现，那必须要修改标准库ostream和istream,到这两个库中去成员函数重载，显然这是不可以的，那就只能用友元重载实现了。还是举个例子来实现：

class Complex{
private:
float _x;
float _y;
public:
Complex(float x = 0, float y = 0):_x(x), _y(y){}
friend ostream& operator <<(ostream& os, const Complex& a){
os << "( " << a._x << "," << a._y << " )";
return os;
}
friend istream& operator >>(istream& is, Complex& a){
is >> a._x;
is >> a._y;
return is;
}
};

int main()
{
Complex c;
cin >> c;
cout << c << endl;
return 0;
}

这里注意两点，一个是友元重载返回值类型，一个是参数，就不多解释了，与前面的双目运算符重载大同小异。

这里要多说一个例子，对于自己编写的string类型的流运算符的重载：

class MyString
{
public:
MyString(char *str = "");
~MyString();
friend ostream& operator <<(ostream& os, const MyString &s);
friend istream& operator >>(istream& is, MyString  &s);
private:
char *_str;
};

MyString::MyString(char *c)
{
int len = strlen(c);
_str = new char[len + 1];
strcpy(_str, c);
}

MyString::~MyString() {
delete[]_str;
}

ostream& operator <<(ostream& os, const MyString &s) {
os << s._str;
return os;
}

istream& operator >>(istream& is, MyString &s) {
is >> s._str;
return is;
}

流输出运算符重载没什么好说的，重点看一下流输入运算符，首先要知道直接对一个char 类型进行直接输入是有危险的， 因为不知道要输入多少个字符，那系统就不知道开辟多少个空间，有可能空间不够，有可能空间浪费，这都是不可取的，那就用一个固定空间大小的char buf[BUFSIZ];来代替char 的输入,代码修改一下：

istream& operator >>(istream& is, MyString &s) {
delete[]s._str;
char buf[BUFSIZ];
is >> buf;
int len = strlen(buf);
s._str = new char[len + 1];
strcpy(s._str, buf);
return is;
}

这样就保证了一种安全性，这是字符串流输入重载需要注意的一点。