Перегрузка оператора ввода
Перегрузка оператора ввода (>>) похожа на перегрузку оператора вывода, но, к сожалению, возможностей для ошибок гораздо больше. Вот, например, его реализация для класса 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" в "хорошем" состоянии |
Если объект 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.