source: downloads/ogre/Tools/VRMLConverter/vrmllib/src/commentstream.cpp @ 24

#include "commentstream.h"

namespace utillib {

class commentstreambuf : public std::streambuf {
        std::streambuf * const m_sbuf;
        char m_buf[512];
        const int m_start;
        const int m_end;
        bool m_comment;
public:
        commentstreambuf(std::streambuf *, char start, char end);

        int underflow();
};

commentstreambuf::commentstreambuf(std::streambuf *sb, char start, char end)
        :       m_sbuf(sb), m_start(static_cast<unsigned char>(start)),
                m_end(static_cast<unsigned char>(end)), m_comment(false)
{
        setg(m_buf, m_buf, m_buf);
}

int commentstreambuf::underflow()
{
        char *p = m_buf;
        int ch;
        while (p != m_buf + sizeof m_buf && (ch = m_sbuf->sgetc()) != EOF) {
                m_sbuf->sbumpc();

                if (ch == m_start) m_comment = true;
                else if (ch == m_end) m_comment = false;

                if (!m_comment)
                        *p++ = ch;
        }

        if (p == m_buf)
                return EOF;

        setg(m_buf, m_buf, p);
        return *m_buf;
}

commentstream::commentstream(std::istream &s, char start, char end)
        :       std::basic_istream(new commentstreambuf(s.rdbuf(), start, end))
{
}

commentstream::~commentstream()
{
        delete rdbuf();
}

} // namespace utillib