stream.cc

#include <vtestbed/iges/stream.hh>

#include <iostream>

void
vtb::iges::iges_ostreambuf::section(Section s) {
    if (s == this->_section) {
        return;
    }
    this->finish();
    this->_section = s;
    this->_maxsize = (s == Section::ParameterData) ? 64 : 72;
    this->_lineno = 0;
    auto offset = pptr() - pbase();
    reset();
    pbump(offset);
}

auto
vtb::iges::iges_ostreambuf::overflow(int_type ch) -> int_type {
    newline();
    if (!traits_type::eq_int_type(ch, traits_type::eof())) {
        *pptr() = ch;
        pbump(1);
    }
    return ch;
}

std::streamsize
vtb::iges::iges_ostreambuf::xsputn(const char_type* first, std::streamsize n) {
    if (epptr()-pptr() < n && n <= max_size()) {
        newline();
    }
    auto old = first;
    auto last = first + n;
    while (first != last) {
        auto m = std::min(epptr()-pptr(), last-first);
        traits_type::copy(pptr(), first, m);
        first += m;
        pbump(m);
        if (pptr() == epptr()) {
            newline();
        }
    }
    return first - old;
}

auto
vtb::iges::iges_ostreambuf::sync() -> int_type {
    this->finish();
    return this->_sink->pubsync();
}

void
vtb::iges::iges_ostreambuf::newline() {
    auto n = this->epptr() - this->pptr();
    if (n > 0) {
        traits_type::assign(this->pptr(), n, ' ');
    }
    auto line_no = line_number();
    if (this->_section == Section::ParameterData) {
        std::snprintf(
            this->epptr(),
            2*8+2,
            "%8zd%c%7zd\n",
            2*this->_number - 1,
            section_character(),
            line_no
        );
    } else {
        std::snprintf(
            this->epptr(),
            8+2,
            "%c%7zd\n",
            section_character(),
            line_no
        );
    }
    this->_sink->sputn(this->data(), 81);
    reset();
    ++this->_lineno;
}