split.h

Go to the documentation of this file.

#ifndef _utl_split_h_
#define _utl_split_h_

#include <string>


namespace utl {

  template<class istream>
  class split {

  public:
    class iterator : public std::iterator<std::input_iterator_tag, std::string> {
    public:
      typedef iterator self_type;
      typedef const std::string value_type;
      typedef const std::string& reference;
      typedef const std::string* pointer;
      typedef std::input_iterator_tag iterator_category;

      iterator() { } // = also end iterator
      iterator(istream& is, const char delim = '\n') : _istream_ptr(&is), _delim(delim) { next(); }
      iterator& operator++() { next(); return *this; }
      const std::string& operator*() const { return _line; }
      const std::string* operator->() const { return &operator*(); }

      bool
      operator==(const iterator& it)
        const
      {
        return
          _istream_ptr == it._istream_ptr &&
          _delim == it._delim &&
          _line == it._line;
      }

      bool operator!=(const iterator& it) const { return !operator==(it); }

    private:
      void
      next()
      {
        if (!_istream_ptr)
          return;
        if (!std::getline(*_istream_ptr, _line, _delim)) {
          _istream_ptr = nullptr;
          _delim = '\0';
          _line.clear();
        }
      }

      istream* _istream_ptr = nullptr;
      char _delim = '\0';
      std::string _line;
    };

    split(istream& is, const char delim = '\n') : _istream(is), _delim(delim) { }

    iterator begin() { return iterator(_istream, _delim); }
    iterator end() { return iterator(); }

    operator std::vector<std::string>()
    {
      return std::vector<std::string>(begin(), end());
    }

  private:
    istream& _istream;
    const char _delim;
  };


  template<class istream>
  inline
  split<istream>
  make_split(istream& is)
  {
    return split<istream>(is);
  }

}


#endif