打破重载的提取运算符? (C ++)

时间:2010-02-10 03:01:40

标签: c++ operator-overloading

我正在尝试使用重载的“>>”扫描文件中的输入。

问题是,我不知道如何处理文件结尾。 在这种情况下,我的文件由一个数字组成,后跟几个字符

例如:

9rl

8D

6FF

istream &operator>>(istream &is, Move &move)
{
  char c;
  int i = 0;

  c = is.get();

  if (!isalnum(c))
      return;

  move.setNum(c); // I convert the char to an int, but I'l edit it out

  while ( (c = is.get()) != '\n')
  {
    move.setDirection(i, c); //sets character c into in array at index i
    i++;

  } // while chars are not newline

  return is;
} // operator >>

当我将此字符作为常规函数时,字符为字母数字的测试工作,但在此处不起作用,因为它期望返回输入流。我也尝试过返回NULL。建议?

编辑:这是在while循环中调用的,所以我试图找出一些方法来触发一些标志,以便我可以突破循环。在我之前的函数中,我返回一个布尔值,如果成功则返回true,如果字符不是字母数字则返回false

2 个答案:

答案 0 :(得分:2)

返回is。呼叫者应检查流是否有错误。

请务必根据需要设置错误位:

std::istream &operator>>(std::istream &is, Move &move)
{
  char c;
  int i = 0;

  c = is.get();
  if (is.eof())
    return is;
  else if (c < '0' || c > '9') {
    is.setstate(std::ios::badbit);
    return is;
  }
  else
    move.setNum(c-'0');

  while ( (c = is.get()) != '\n' && is)
    move.setDirection(i++, c);

  if (c != '\n')
    is.setstate(std::ios::badbit);
  return is;
}

使用它,如下所示:

int main(int argc, char **argv)
{
  std::stringstream s;

  s << "9rl\n"
    << "8d\n"
    << "6ff\n";
  s.seekg(0);

  Move m;
  while (s >> m)
    std::cout << m;

  if (s.bad())
    std::cerr << argv[0] << ": extraction failed\n";

  return 0;
}

请注意,代码仅在成功提取后才使用实例m

答案 1 :(得分:2)

您可以使用ios::bad将流的标记设置为ios::failios::setstate等状态。这将允许调用者测试流,或者在为流启用异常的情况下,将引发异常。

您也无法检查流的状态。 C ++ FAQ lite有一个great section来解释这一点。为了澄清这一点,我在下面添加了代码片段。

c = is.get();
// the stream has not been tested to see if it read into c correctly
if (!isalnum(c))
    return;