当前位置:主页 > 软件编程 > C代码 >

C++ 流插入和流提取运算符的重载的实现

时间:2021-04-18 09:51:27 | 栏目:C代码 | 点击:

01 流插入<<运算符的重载

C++ 在输出内容时,最常用的方式:

std::cout << 1 <<"hello";

问题:

原因:

对于std::cout << 1 <<"hello";这条语句,有可能按以下的方式重载成 ostream 类的成员函数:

ostream & ostream::operator<<(int n)
{
  .... // 输出n整型的代码
  return *this;
}

ostream & ostream::operator<<(const char * s)
{
  .... // 输出s字符串的代码
  return *this;
}

02 流插入<<运算符重载的例子

假定我们要想把某个对象里的内容进行打印输出,那么我们可以重载 ostream 类的流插入 << 运算符。

下面以 CStudent 类作为例子:

class CStudent // 学生类
{
public:
  // 构造函数
  CStudent(int id = 0, int age = 0, string name = ""):m_id(id), m_age(age), m_name(name) { }
  
  // 将该函数声明成友元函数
  // 目的是使得函数可以访问CStudent类的私有成员变量
  friend ostream & operator<<(ostream & o, const CStudent & s);
  
private:
  int m_age;   // 年龄
  int m_id;    // ID号
  string m_name; // 名字
};

// 重载ostream对象的流插入<<运算符函数
// 目的是使得能打印输出CStudent对象的信息
ostream & operator<<(ostream & o, const CStudent & s)
{
  o << s.m_id << "," << s.m_age << "," << s.m_name;
  return o;
}

int main()
{
  CStudent stu(1, 20, "小林coding");
  std::cout << stu ; // 输出std对象的全部信息
  
  return 0;
}

输出结果:

1,20,小林coding

需要注意是 ostream & operator<<(ostream & o, const CStudent & s) 函数是全局的,所以函数的第一个参数必须要传入 ostream 的对象,并且 CStudent 类需要将此函数声明成友元函数,使得函数可以访问 CStudent 类的私有成员变量。

03 流提取>>运算符重载的例子

还是以 CStudent 类作为例子,假设想通过键盘的输入的内容,来初始化对象,则我们可以重载 istream 类的流提取 >> 运算符。

class CStudent // 学生类
{
public:

  // 构造函数
  CStudent(int id = 0, int age = 0, string name = ""):m_id(id), m_age(age), m_name(name) { }
  
  // 将该函数声明成友元函数
  // 目的是使得函数可以访问CStudent类的私有成员变量
  friend ostream & operator<<(ostream & o, const CStudent & s);
  
  // 将该函数声明成友元函数
  // 目的是使得函数可以给CStudent类的私有成员变量进行赋值
  friend istream & operator>>(istream & is, CStudent & s);
  
private:
  int m_age;   // 年龄
  int m_id;    // ID号
  string m_name; // 名字
};

// 重载ostream对象的流插入<<运算符函数
// 目的是使得能打印输出CStudent对象的信息
ostream & operator<<(ostream & o, const CStudent & s)
{
  o << s.m_id << "," << s.m_age << "," << s.m_name;
  return o;
}

// 重载istream对象的流提取>>运算符函数
// 目的是使得初始化CStudent对象的内容
istream & operator>>(istream & is, CStudent & stu)
{
  string inputStr;
  is >> inputStr;
  
  int pos = inputStr.find(",", 0);     // 查找首次出现逗号的位置
  string tmpStr = inputStr.substr(0, pos); // 截取从0到pos位置的字符串
  stu.id = atoi(tmpStr.c_str());      // atoi可以将char*类型的内容转成int类型
  
  int pos2 = inputStr.find(",", pos + 1);      // 查找第二次出现逗号的位置
  tmpStr = inputStr.substr(pos + 1, pos2 - pos -1); // 取出age的值
  stu.age = atoi(tmpStr.c_str());          // atoi可以将char*类型的内容转成int类型
  
  tmpStr = inputStr.substr(pos2 + 1, inputStr.length() - pos2 - 1); // 取出name的值
  stu.name = tmpStr;
  
  return is;
}

int main()
{
  CStudent stu;
  
  // 将输入的信息,初始化stu对象
  cin << stu;
  
  // 输出std对象的信息
  cout >> stu;
  
  return 0;
}

输入内容和输出内容:

// 输入内容:
1,20,小林coding

// 输出内容:
1,20,小林coding

04 小结

要想流插入 << 运算符和流提取 >> 运算符能针对自定义的对象,那么我们就需要重载针对该对象的 ostream 类的 << 运算符 和 istream 的 >> 运算符,并且只能重载成全局的函数,然后在 CStudent 类里需要把上面的两个重载函数声明成友元函数,使得两个重载的函数可以访问和赋值 CStudent 类里的私有成员函数。

您可能感兴趣的文章:

相关文章