Перегрузка оператора istream >> c ++ - PullRequest
1 голос
/ 27 июня 2011

Скажем, у меня есть вектор символов, и я поместил его в поток как строку, а не как вектор символов. Как бы я мог вернуть вектор символов с помощью оператора >>?

class C{
    private:
        vector<char> c;

    public:
        C(string str){
          for(int x = 0; x < str.size(); x++)
              c.push_back(str[x]);
        }

        vector<char> data(){
           return c;
        }       
};

ostream operator<<(ostream & stream, C & in){
   for(int x = 0; x < in.data().size(); x++)
      stream << in.data()[x];
   return stream;
}

istream operator>>(istream & stream, C & in){
    // ???
    // what kind of loop?
}

Ответы [ 5 ]

3 голосов
/ 27 июня 2011

Я бы написал ваш пример так ...

#include <algorithm>
#include <iterator>
#include <vector>
#include <iostream>
#include <sstream>

class C
{
    std::vector<char> mData;

  public:
    // Constructor, note im using the iterator range 
    // vector constructor.
    C(const std::string& str)
    : mData(str.begin(), str.end())
    {
    }

    // Returning data by const reference to avoid 
    // copying a potentially large object.
    const std::vector<char>& data() const
    {
        return mData;
    }

    // declared the input operator as a friend as I need it to
    // access mData - alternatively you could write a "loadFromStream"
    // type function that does the same, declare this as non-friend, and call that.
    friend std::istream& operator>>(std::istream& is, C& c);
};

std::ostream& operator<<(std::ostream& os, const C& c)
{
    // Use an ostream_iterator to handle output of the vector
    // using iterators.
    std::copy(c.data().begin(), 
              c.data().end(), 
              std::ostream_iterator<char>(os, ""));

    return os;
}

std::istream& operator>>(std::istream& is, C& c)
{
    // load the data using the assign function, which
    // clears any data already in the vector, and copies 
    // in the data from the specified iterator range.
    // Here I use istream_iterators, which will read to the end
    // of the stream.  If you dont want to do this, then you could 
    // read what you want into a std::string first and assign that.
    c.mData.assign(std::istream_iterator<char>(is),
                   std::istream_iterator<char>());

    return is;
}

int main()
{
    C c("Hello");

    std::stringstream ss;
    ss << c;

    std::cout << ss.str() << std::endl;

    C d("");
    ss >> d;

    std::cout << d.data().size() << std::endl;

    return 0;
}
2 голосов
/ 27 июня 2011

Вы всегда можете построить одно из другого:

std::vector<char> A = getVector();
std::string       B = getString();

std::vector<char> newB(B.begin(), B.end());
std::string       newA(A.begin(), A.end());

Используя это, вы сможете написать свой оператор in-stream, например,

std::string stmp;
stream >> stmp;
std::vector<char> vtmp(stmp.begin(), stmp.end());
c.swap(vtmp);
1 голос
/ 27 июня 2011

Вы можете использовать итератор istream. Его конструктор по умолчанию инициализируется до конца потока.

http://www.cplusplus.com/reference/std/iterator/istream_iterator/

Ваш код будет выглядеть примерно так.

typedef std::istream_iterator<char> charstream_it;
c = std::vector<char>(charstream_it(stream),  charstream_it());

В вашем конструкторе вам, вероятно, следует использовать конструктор копирования в стиле итератора STL, как рекомендовано Kerrek SB.

C::C(string str):
c(str.begin(), str.end()) {};
1 голос
/ 27 июня 2011

Прежде всего, вы захотите возвращать свои потоки в виде ссылок, а не по значению.Кроме того, data() должен возвращать константную ссылку на вектор, чтобы он не копировался (важно, если это большой вектор).

Что касается перегрузки >>, я бы попробовал что-то вроде:

istream& operator>>(istream& stream, C& in) {
    for (char c; /* some end condition */;) {
        stream >> c;
        in.c.push_back(c);
    }
}

Конечно, этот подход потребует объявления operator>>(istream&, C&) функции друга.Альтернативой является предоставление эквивалента функции append(char) вашему общедоступному интерфейсу.Кроме того, data() должен быть помечен как const, так что вся подпись будет const vector<char>& data() const, давая понять, что это строго метод доступа.

1 голос
/ 27 июня 2011

Чтобы сделать то, что >> for std::string, вам действительно нужно просто использовать std::string.Обратите внимание, что вам нужно передать C по ссылке, а не по значению, и операторы должны вернуть исходный поток по ссылке.Кроме того, я не могу не думать, что использование std::vector<char> вместо std::string не очень полезно (плюс, ctor неэффективен - по крайней мере, резервируйте str.length(), если вы делаете это).

istream& operator>>(istream& s, C& in) {
    std::string tmp;
    s >> tmp;
    // you also need to return reference in in.data() for this to work
    // or declare operator>> as friend and use in.c directly
    in.data().assign(tmp.begin(), tmp.end());
    return s;
}
...