背景
我们很多时候看到类似以下代码
int main(int argc, char **argv)
{
int value = 10;
cout << value << endl;
return 0;
}
输出:
但如果我们自定义的类,是否也能这样的输出呢,看以下代码
很显然,这已经报错了
解决方法
这时候就要引进我们c++的重载机制了
#include <iostream>
using namespace std;
class MyClass
{
public:
MyClass() {}
~MyClass() {}
friend ostream& operator<<(ostream& out, MyClass& a); //添加这句
private:
int _value = 10;
};
//函数实现
ostream& operator<<(ostream& out, MyClass& a)
{
out << a._value;
return out;
}
int main(int argc, char **argv)
{
MyClass a;
cout << a << endl;
return 0;
}
输出结果
可以看到已经输出正确结果了
那既然有了输出 << ,又想有输入>>怎么办?很简单,继续重载>>即可
完整代码
#include <iostream>
using namespace std;
class MyClass
{
public:
MyClass() {}
~MyClass() {}
friend ostream& operator<<(ostream& out, MyClass& a);
friend istream& operator>>(istream& in, MyClass& a);
private:
int _value = 10;
};
ostream& operator<<(ostream& out, MyClass& a)
{
out << a._value;
return out;
}
istream& operator>>(istream& in, MyClass& a)
{
in >> a._value;
return in;
}
int main(int argc, char **argv)
{
MyClass a;
cout << "输入:";
cin >> a;
cout << endl << "输出:" << a << endl;
return 0;
}
输出
结束
使用关键字operator,记得需要用友元函数方式去重载,不使用成员函数去重载!