LineNumberStream.hpp

Go to the documentation of this file.

#ifndef LINE_NUMBER_STREAM_DEFINED
#define LINE_NUMBER_STREAM_DEFINED
 
#include <string>
 
namespace sisi4s {
class LineNumberStream {
public:
  LineNumberStream(std::istream *stream_,
                   std::string const &source_,
                   int const tabWidth_ = 2)
      : line(1)
      , column(1)
      , tabWidth(tabWidth_)
      , stream(stream_)
      , source(source_) {}
  ~LineNumberStream() { delete stream; }
  int peek() { return stream->peek(); }
  int get() {
    int character(stream->get());
    switch (character) {
    case '\n':
      ++line;
      column = 1;
      break;
    case '\t': column += tabWidth; break;
    default: ++column;
    }
    return character;
  }
  std::istream *getStream() { return stream; }
  std::string getSource() { return source; }
  int getLine() { return line; }
  int getColumn() { return column; }
 
protected:
  int line, column, tabWidth;
  std::istream *stream;
  std::string source;
};
} // namespace sisi4s
 
#endif