C++的流插入运算符"<<"和流提取运算符“>>"是C++编译系统在类库中提供的,所有C++编译编译系统都在其类库中提供输入流类istream和输出流类ostream,cin和cout分别是
istream和ostream类的对象。
对“<<"和“>>"重载的函数形式如下:
istream & operator>>(istream &,自定义类 &);
ostream & operator<<(ostream & 自定义类 &);
重载>>的函数的第一个参数和函数的类型都必须是istream&类型,也就是istream类对象的引用,第二个参数是要进行输入操作的类。
重载<<的函数的第一个参数和函数的类型都必须是ostream&类型,也就是ostream类对象的引用,第二个参数是要进行输出操作的类。
因此,只能将重载“>>"和“<<"的函数作为友元函数,而不能将他们定义为成员函数。
同样使用上次的复数类作为例子:
#include<iostream>
using namespace std;
class Complex
{
public:
Complex(){real = 0;imag = 0;}
Complex(double r,double i){real =r;imag = i;};
Complex operator + (Complex& c2); //声明重载运算符“+”的函数
friend ostream& operator <<(ostream&,Complex&); //声明友元重载运算符“<<"函数
friend istream& operator >>(istream&,Complex&); //声明友元重载运算符“>>”函数
void display();
private:
double real;
double imag;
};
Complex Complex::operator + (Complex& c2) //定义重载运算符“+”的函数
{
Complex c;
c.real = real + c2.real;
c.imag = imag + c2.imag;
return c;
}
void Complex::display()
{
cout<<"real = "<<real<<endl<<"imag = "<<imag<<endl;
}
ostream& operator << (ostream& output,Complex& c)
{
output<<"("<<c.real;
if (c.imag >= 0)output<<"+";
output<<c.imag<<"i)"<<endl;
return output;
}
istream& operator >> (istream& input,Complex& c)
{
cout<<"input real part and imaginary part of complex number:";
input>>c.real>>c.imag;
return input;
}
int main()
{
Complex c1,c2;
cin>>c1>>c2;
cout<<"c1="<<c1<<endl;
cout<<"c2="<<c2<<endl;
return 0;
}
以运算符重载“<<"为例,运算符“<<"的左面是cout,它是在头文件iostream中声明的ostream类对象,“<<"右面是一个自定义类的对象。
所以编译系统把cout<<c3解释为
operator<<(cout,c3)
注意上面的return output 语句,它实现了连续向输出流插入信息。output是cout的别名,因此,return output就是return cout。
同时注意重载<<时考虑了当输入为负值的情况,这对于程序的完整性和可读性有重要影响。