重载<<使用ostream和枚举类型

时间:2017-12-13 13:45:22

标签: c++ oop enums operator-overloading

以下是我的代码。我正在尝试打印一个枚举数据类型变量,一个直接使用其名称,另一个使用函数返回与原始变量相同的数据类型。

#include <iostream>

using namespace std;
enum days{SUN, MON, TUE, WED, THU, FRI, SAT};


inline days dayaftertomorrow(days d)
{
  return static_cast<days>((static_cast<int>(d) + 2) % 7);
}

ostream& operator<< (ostream& out, days& day)
{
  switch(day)
  {
    case SUN: out << "SUN"; break;
    case MON: out << "MON"; break;
    case TUE: out << "TUE"; break;
    case WED: out << "WED"; break;
    case THU: out << "THU"; break;
    case FRI: out << "FRI"; break;
    case SAT: out << "SAT"; break;
  }
  return out;
}


int main(int argc, char const *argv[]) {
  days d = MON, e;
  e = dayaftertomorrow(d);
  cout << d << "\t" << e << endl;
  cout << d << "\t" << dayaftertomorrow(d) << endl;
  return 0;
}

我希望结果为

MON     WED
MON     WED

但我得到了

MON     WED
MON     3

我该如何解决这个问题?

1 个答案:

答案 0 :(得分:4)

请注意,operator<<通过引用非const来获取参数daydayaftertomorrow按值返回; dayaftertomorrow返回的是一个临时的,不能绑定引用非const。然后,对于cout << dayaftertomorrow(d),您的operator<<将不会被调用,枚举器将implicitly converted改为int,然后通过std::basic_ostream::operator<<打印出来;这就是你3的原因。

要解决此问题,您可以更改参数类型以引用const;哪个可以绑定到临时的。或者将其更改为按值传递。 e.g。

ostream& operator<< (ostream& out, const days& day)
//                                 ~~~~~

ostream& operator<< (ostream& out, days day)

LIVE