С++ для начинающих

       

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


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


    Содержание раздела