MegEngine/lite/load_and_run/src/helpers/text_table.cpp

#include "text_table.h"

using namespace mgb;

namespace {
inline void mid(std::ostream& os, const std::string& str, size_t max_w) {
    size_t l = (max_w - str.length()) / 2 + str.length();
    size_t r = max_w - l;
    os << std::setw(l) << std::right << str;
    if (r > 0)
        os << std::setw(r) << ' ';
}
inline size_t char_length(char c) {
    return c ? 1 : 0;
}
}  // namespace

void TextTable::adjuster_last_row() {
    if (m_rows.empty())
        return;
    auto& row = m_rows.back();
    if (row.params.horizontal == 0 or row.params.vertical == 0) {
        row.params.corner = 0;
    }
    if (row.params.horizontal != 0 && row.params.vertical != 0 &&
        row.params.corner == 0) {
        row.params.corner = row.params.horizontal;
    }
}

void TextTable::show(std::ostream& os) {
    if (m_rows.empty())
        return;
    auto& last_row = m_rows.front();
    bool first = true;
    for (auto& row : m_rows) {
        auto& lrow =
                (last_row.values.size() * char_length(last_row.params.horizontal)) >
                                (row.values.size() * char_length(row.params.horizontal))
                        ? last_row
                        : row;
        // line before row
        if (lrow.params.horizontal) {
            if (not first)
                os << std::endl;
            os << m_prefix;
            if (lrow.params.corner)
                os << lrow.params.corner;
            size_t skip_size = 0;
            // table name
            if (first) {
                os << m_name;
                skip_size = m_name.length();
            }
            for (size_t i = 0; i < lrow.values.size(); ++i) {
                auto max_w = m_cols_max_w.at(i) + m_padding * 2;
                if (max_w + char_length(lrow.params.corner) <= skip_size) {
                    skip_size = skip_size - max_w - char_length(lrow.params.corner);
                    continue;
                }
                size_t rest = max_w + char_length(lrow.params.corner) - skip_size;
                skip_size = 0;
                if (rest > char_length(lrow.params.corner)) {
                    os << std::string(
                            rest - char_length(lrow.params.corner),
                            lrow.params.horizontal);
                    rest = char_length(lrow.params.corner);
                }
                if (rest > 0 && lrow.params.corner)
                    os << lrow.params.corner;
            }
        } else if (first) {
            os << m_prefix << ' ' << m_name;
        }
        first = false;
        os << std::endl << m_prefix;
        if (row.params.vertical)
            os << row.params.vertical;
        // row
        for (size_t i = 0; i < row.values.size(); ++i) {
            auto& str = row.values.at(i);
            auto max_w = m_cols_max_w.at(i) + 2 * m_padding;
            if (row.params.align == Align::Mid) {
                mid(os, str, max_w);
            } else if (row.params.align == Align::Left) {
                os << std::setw(max_w) << std::left << str;
            } else {
                os << std::setw(max_w) << std::right << str;
            }
            if (row.params.vertical)
                os << row.params.vertical;
        }
        last_row = row;
    }
    if (last_row.params.horizontal) {
        os << std::endl << m_prefix;
        if (last_row.params.corner)
            os << last_row.params.corner;
        for (size_t i = 0; i < last_row.values.size(); ++i) {
            auto max_w = m_cols_max_w.at(i);
            std::string tmp(max_w + m_padding * 2, last_row.params.horizontal);
            os << tmp;
            if (last_row.params.corner)
                os << last_row.params.corner;
        }
    }
}