20.5. Перегрузка оператора ввода

Перегрузка оператора ввода (>>) похожа на перегрузку оператора вывода, но, к сожалению, возможностей для ошибок гораздо больше. Вот, например, его реализация для класса WordCount:

#include <iostream>

#include "WordCount.h"

/* необходимо модифицировать определение класса WordCount, чтобы

оператор ввода был другом

class WordCount {

friend ostream& operator<<( ostream&, const WordCount& );

friend istream& operator>>( istream&, const WordCount& );

*/

istream&

operator >>( istream &is, WordCount &wd )

{

/* формат хранения объекта WordCount:

* <2> строка

* <7,3> <12,36>

*/

int ch;

/* прочитать знак '<'. Если его нет,

* перевести поток в ошибочное состояние и выйти

*/

if ((ch = is.get()) != '<' )

{

// is.setstate( ios_base::badbit );

return is;

}

// прочитать длину

int occurs;

is >> occurs;

// читать до обнаружения >; ошибки не контролируются

while ( is && (ch = is.get()) != '>' ) ;

is >> wd._word;

// прочитать позиции вхождений;

// каждая позиция имеет формат: < строка, колонка >

for ( int ix = 0; ix < occurs; ++ix )

{

int line, col;

// извлечь значения

while (is && (ch = is.get())!= '<' ) ;

is >> line;

while (is && (ch = is.get())!= ',' ) ;

is >> col;

while (is && (ch = is.get())!= '>' ) ;

wd._occurList.push_back( Location( line, col ));

}

return is;

}

На этом примере показан целый ряд проблем, имеющих отношение к возможным ошибочным состояниям входного потока:

• поток, чтение из которого невозможно из-за неправильного формата, переводится в состояние fail:

is.setstate( ios_base::failbit );

• операции вставки и извлечения из потока, находящегося в ошибочном состоянии, не работают:

while (( ch = is.get() ) != lbrace)

Инструкция зациклится, если объект istream будет находиться в ошибочном состоянии. Поэтому перед каждым обращением к get() проверяется отсутствие ошибки:

// проверить, находится ли поток "is" в "хорошем" состоянии

while ( is && ( ch = is.get() ) != lbrace)

Если объект istream не в “хорошем” состоянии, то его значение будет равно false. (О состояниях потока мы расскажем в разделе 20.7.)

Данная программа считывает объект класса WordCount, сохраненный оператором вывода из предыдущего раздела:

#include <iostream>

#include "WordCount.h"

int main()

{

WordCount readIn;

// operator>>( cin, readIn )

cin >> readIn;

if ( !cin ) {

cerr << "Ошибка ввода WordCount" << endl;

return -1;

}

// operator<<( cout, readIn )

cout << readIn << endl;

}

Выводится следующее:

<10> rosebud

<11,3> <11,8> <14,2> <34,6> <49,7> <67,5>

<81,2> <82,3> <91,4> <97,8>

Упражнение 20.9

Оператор ввода класса WordCount сам читает объекты класса Location. Вынесите этот код в отдельный оператор ввода класса Location.

Упражнение 20.10

Реализуйте оператор ввода для класса Date из упражнения 20.7 в разделе 20.4.

Упражнение 20.11

Реализуйте оператор ввода для класса CheckoutRecord из упражнения 20.8 в разделе 20.4.