Потоковый вопрос об операторе C++

Я предполагаю, что это могло бы быть простым вопросом для всех гуру здесь, но я так или иначе не мог выяснить ответ.

Я хочу быть в состоянии записать csv ячейки в поток, настолько простой:

stream << 1 << 2 << "Tom" << std::endl;

который создал бы вывод как 1,2, Tom. Как я могу достигнуть этого? Я полагал, что должен создать пользовательский streambuf (поскольку я не думаю, что это - правильный способ сделать это на потоковом уровне, это была бы реальная боль только для перегрузки <<для всех типов), но я не уверен, как <<обычно реализуется. Это называет помещенным или пишет или что. Я должен переопределить их или что? Или я просто пропускал что-то полностью?

Я ценил бы любую справку:)

С наилучшими пожеланиями,

10
задан Tom 4 March 2010 в 02:07
поделиться

3 ответа

Добиться чего-то вроде 98% пути туда не так уж сложно:

#include <iostream>

class add_comma { 
    std::ostream &os;
    bool begin;
    typedef add_comma &ref;
public:
    add_comma(std::ostream &o) : os(o), begin(true) {}

    template <class T>
    ref operator<<(T const &t) { 
        if (!begin)
            os << ",";
        os << "\"" << t << "\"";
        begin = false;
        return *this;
    }

    ref operator<<(std::ostream &manip(std::ostream &o) ) {
        if (&manip == &std::endl)
            reset();
        manip(os);
        return *this;
    }

    void reset() { begin = true; }

    operator void *() { return (void *)os; }
};

int main() { 
    add_comma a(std::cout);

    a << 1 << 2 << "This is a string" << std::endl;
    a << 3 << 4 << "Another string" << std::endl;
    return 0;
}

Edit: Я исправил код по крайней мере до некоторой степени - теперь он ставит запятые только между элементами, которые записываются, а не в начале строки. Однако он распознает только "endl" как сигнал начала новой записи - новая строка в строковом литерале, например, не будет работать.

10
ответ дан 3 December 2019 в 22:36
поделиться

Хотя я могу оценить идею перегрузки оператора потока, я бы поставил под сомнение практику решения данной проблемы.

1. Объектно-ориентированный подход

Если вы хотите писать в файл .csv , то каждая строка, вероятно, должна иметь тот же формат, что и другие? К сожалению, ваш оператор потока не проверяет это.

Я думаю, что вам нужно создать объект Line , который будет потоковым, и будет проверять каждое поле перед записью их в файл (и записывать их в правильном формате). Хотя это и не так модно, у вас будет гораздо больше шансов добиться надежной реализации здесь.

Предположим, что (например) вы хотите вывести 2 целых числа и строку:

class Line
{
public:
  Line(int foo, int bar, std::string firstName):
    mFoo(foo), mBar(bar), mFirstName(firstName)

  friend std::ostream& operator<<(std::ostream& out, const Line& line)
  {
    return out << line.mFoo << ',' << line.mBar << ','
               << line.mFirstName << std::endl;
  }
private:
  int mFoo;
  int mBar;
  std::string mFirstName;
};

И использование этого остается очень простым:

std::cout << Line(1,3,"Tom") << Line(2,4,"John") << Line(3,5,"Edward");

2. Хотите повеселиться?

Это может показаться скучным, и вы можете захотеть поиграть, но при этом иметь некоторый контроль над написанным ... ну, позвольте мне ввести мета-программирование шаблонов в бой;)

Здесь это предполагаемое использование:

// Yeah, I could wrap this mpl_::vector bit... but it takes some work!
typedef CsvWriter< mpl_::vector<int,int,std::string> > csv_type;

csv_type(std::cout) << 1 << 3 << "Tom" << 2 << 4 << "John" << 3 << 5 << "Edward";

csv_type(std::cout) << 1 << 2 << 3; // Compile Time Error:
                                    // 3 is not convertible to std::string

Это было бы интересно, не так ли? Он отформатировал бы строку и обеспечил бы меру проверки ... Всегда можно усложнить дизайн, чтобы он делал больше (например, регистрировал валидаторы для каждого поля или для всей строки и т. Д.), Но это уже достаточно сложно.

// namespace mpl_ = boost::mpl

/// Sequence: MPL sequence
/// pos: mpl_::size_t<N>, position in the Sequence

namespace result_of {
  template <class Sequence, class pos> struct operator_in;
}

template < class Sequence, class pos = mpl_::size_t<0> >
class CsvWriter
{
public:
  typedef typename mpl_::at<Sequence,pos>::type current_type;
  typedef typename boost::call_traits<current_type>::param_type param_type;

  CsvWriter(std::ostream& out): mOut(out) {}

  typename result_of::operator_in<Sequence,pos>::type
  operator<<(param_type item)
  {
    typedef typename result_of::operator_in<Sequence,pos>::type result_type;

    if (pos::value != 0) mOut << ',';
    mOut << item;

    if (result_type::is_last_type::value) mOut << std::endl;              

    return result_type(mOut);
  }

private:
  std::ostream& mOut;
}; // class CsvWriter


/// Lil' bit of black magic
namespace result_of { // thanks Boost for the tip ;)

  template <class Sequence, class pos>
  struct operator_in
  {
    typedef typename boost::same_type<
        typename mpl_::size<Sequence>::type,
        typename mpl_::next<pos>::type
      > is_last_type;

    typedef typename mpl_::if_<
      is_last_type,
      CsvWriter< Sequence, mpl_::size_t<0> >,
      CsvWriter< Sequence, typename mpl_::next<pos>::type >
    >::type;
  }; // struct operator_in<Sequence,pos>

} // namespace result_of

Здесь у вас есть средство записи потока, которое гарантирует, что файл cvs правильно отформатирован ... без символов новой строки в строках;)

5
ответ дан 3 December 2019 в 22:36
поделиться

Если A - итератор по элементам...

copy(A, A + N, ostream_iterator<int>(cout, ","));
0
ответ дан 3 December 2019 в 22:36
поделиться
Другие вопросы по тегам:

Похожие вопросы: