重载流插入和流提取运算符

C++的流提取运算符>>和流插入运算符<<能用来输入输出标准类型的数据。这两个运算符是C++编译器在类库中提供的，可以处理包括字符串和内存地址在内的每个标准数据类型。如果我们为自定义类型（C++类）重载这两个运算符，那么他们就也能输入输出该自定义类型。

 

如下是一个简单的例子，用来处理用户自定义的电话号码类PhoneNumber的数据，源码如下：

类文件：PhoneNumber.h

1. #ifndef PHONENUMBER_H
2. #define PHONENUMBER_H
4. #include <iostream>
5. using namespace std;
7. //Calling any one of the potentially unsafe methods such as 'readsome' in the Standard C++ Library 
8. //will result in Compiler Warning (level 1) C4996.
9. #pragma warning(disable:4996)
11. class PhoneNumber
12. {
13. public:
14.     PhoneNumber();
15.     friend ostream &operator<<(ostream &output, const PhoneNumber &num);
16.     friend istream &operator>>(istream& input, PhoneNumber &num);
18. private:
19.     char areaCode[4];           //3位数字的区号或为空
20.     char exchange[4];           //3位数字的电话局号或为空
21.     char line[5];               //4位数字的线路号或为空
22. };
24. PhoneNumber::PhoneNumber()
25. {
26.     memset(areaCode, 0, sizeof(areaCode));
27.     memset(exchange, 0, sizeof(exchange));
28.     memset(line, 0, sizeof(line));
29. }
31. ostream &operator<<(ostream& output, const PhoneNumber &num)
32. {
33.     output<<"("<<num.areaCode<<")"<<num.exchange<<"-"<<num.line;
34.     
35.     //使得能连续执行cout<<a<<b<<c;
36.     return output;
37. }
39. istream &operator>>(istream& input, PhoneNumber &num)
40. {
41.     input.ignore();                                             //跳过（
42.     input.readsome(num.areaCode, sizeof(num.areaCode)-1);       //输入区号
43.     input.ignore();                                             //跳过）
44.     input.readsome(num.exchange, sizeof(num.exchange)-1);       //输入电话局号
45.     input.ignore();                                             //跳过连字符-
46.     input.readsome(num.line, sizeof(num.line)-1);               //输入线路号
48.     //使得能连续执行cin>>a>>b>>c;
49.     return input;
50. }
52. #endif

主程序文件：Main.h

1. #include "PhoneNumber.h"
3. int main()
4. {
5.     cout<<"--------------------------------------------------------------------/n";
6.     cout<<"              重载流插入和流提取运算符演示程序                      /n";
7.     cout<<"  by Loomman, QQ:28077188, MSN: [email protected] QQ裙：30515563  /n";
8.     cout<<"--------------------------------------------------------------------/n/n";
11.     PhoneNumber phone;
13.     cout<<"Enter a phone number in the form (123)456-789: /n";
14.     cin>>phone;
16.     cout<<"The phone number entered was: /n"<<phone<<endl;
18.     int a;
19.     cin>>a;
20.     return a;
21. }

运行结果如下：

--------------------------------------------------------------------
              重载流插入和流提取运算符演示程序
  by Loomman, QQ:28077188, MSN: [email protected] QQ裙：30515563
--------------------------------------------------------------------

Enter a phone number in the form (123)456-789:
(111)222-333
The phone number entered was:
(111)222-333
by Loomman, QQ:28077188, MSN: [email protected] QQ裙：30515563 ☆程序天堂☆ 请尊重作者原创，转载注明来自裂帛一剑博客，谢谢合作。