429 lines
14 KiB
C++
429 lines
14 KiB
C++
|
// Copyright 2012 Google Inc.
|
||
|
|
// All rights reserved.
|
||
|
|
//
|
||
|
|
// Redistribution and use in source and binary forms, with or without
|
||
|
|
// modification, are permitted provided that the following conditions are
|
||
|
|
// met:
|
||
|
|
//
|
||
|
|
// * Redistributions of source code must retain the above copyright
|
||
|
|
// notice, this list of conditions and the following disclaimer.
|
||
|
|
// * Redistributions in binary form must reproduce the above copyright
|
||
|
|
// notice, this list of conditions and the following disclaimer in the
|
||
|
|
// documentation and/or other materials provided with the distribution.
|
||
|
|
// * Neither the name of Google Inc. nor the names of its contributors
|
||
|
|
// may be used to endorse or promote products derived from this software
|
||
|
|
// without specific prior written permission.
|
||
|
|
//
|
||
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||
|
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||
|
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||
|
|
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||
|
|
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||
|
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||
|
|
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||
|
|
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||
|
|
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||
|
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||
|
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
|
|
||
|
|
#include "utils/text/table.hpp"
|
||
|
|
|
||
|
|
#include <algorithm>
|
||
|
|
#include <iterator>
|
||
|
|
#include <limits>
|
||
|
|
#include <sstream>
|
||
|
|
|
||
|
|
#include "utils/sanity.hpp"
|
||
|
|
#include "utils/text/operations.ipp"
|
||
|
|
|
||
|
|
namespace text = utils::text;
|
||
|
|
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
|
||
|
|
/// Applies user overrides to the column widths of a table.
|
||
|
|
///
|
||
|
|
/// \param table The table from which to calculate the column widths.
|
||
|
|
/// \param user_widths The column widths provided by the user. This vector must
|
||
|
|
/// have less or the same number of elements as the columns of the table.
|
||
|
|
/// Values of width_auto are ignored; any other explicit values are copied
|
||
|
|
/// to the output widths vector, including width_refill.
|
||
|
|
///
|
||
|
|
/// \return A vector with the widths of the columns of the input table with any
|
||
|
|
/// user overrides applied.
|
||
|
|
static text::widths_vector
|
||
|
|
override_column_widths(const text::table& table,
|
||
|
|
const text::widths_vector& user_widths)
|
||
|
|
{
|
||
|
|
PRE(user_widths.size() <= table.ncolumns());
|
||
|
|
text::widths_vector widths = table.column_widths();
|
||
|
|
|
||
|
|
// Override the actual width of the columns based on user-specified widths.
|
||
|
|
for (text::widths_vector::size_type i = 0; i < user_widths.size(); ++i) {
|
||
|
|
const text::widths_vector::value_type& user_width = user_widths[i];
|
||
|
|
if (user_width != text::table_formatter::width_auto) {
|
||
|
|
PRE_MSG(user_width == text::table_formatter::width_refill ||
|
||
|
|
user_width >= widths[i],
|
||
|
|
"User-provided column widths must be larger than the "
|
||
|
|
"column contents (except for the width_refill column)");
|
||
|
|
widths[i] = user_width;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return widths;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Locates the refill column, if any.
|
||
|
|
///
|
||
|
|
/// \param widths The widths of the columns as returned by
|
||
|
|
/// override_column_widths(). Note that one of the columns may or may not
|
||
|
|
/// be width_refill, which is the column we are looking for.
|
||
|
|
///
|
||
|
|
/// \return The index of the refill column with a width_refill width if any, or
|
||
|
|
/// otherwise the index of the last column (which is the default refill column).
|
||
|
|
static text::widths_vector::size_type
|
||
|
|
find_refill_column(const text::widths_vector& widths)
|
||
|
|
{
|
||
|
|
text::widths_vector::size_type i = 0;
|
||
|
|
for (; i < widths.size(); ++i) {
|
||
|
|
if (widths[i] == text::table_formatter::width_refill)
|
||
|
|
return i;
|
||
|
|
}
|
||
|
|
return i - 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Pads the widths of the table to fit within a maximum width.
|
||
|
|
///
|
||
|
|
/// On output, a column of the widths vector is truncated to a shorter length
|
||
|
|
/// than its current value, if the total width of the table would exceed the
|
||
|
|
/// maximum table width.
|
||
|
|
///
|
||
|
|
/// \param [in,out] widths The widths of the columns as returned by
|
||
|
|
/// override_column_widths(). One of these columns should have a value of
|
||
|
|
/// width_refill; if not, a default column is refilled.
|
||
|
|
/// \param user_max_width The target width of the table; must not be zero.
|
||
|
|
/// \param column_padding The padding between the cells, if any. The target
|
||
|
|
/// width should be larger than the padding times the number of columns; if
|
||
|
|
/// that is not the case, we attempt a readjustment here.
|
||
|
|
static void
|
||
|
|
refill_widths(text::widths_vector& widths,
|
||
|
|
const text::widths_vector::value_type user_max_width,
|
||
|
|
const std::size_t column_padding)
|
||
|
|
{
|
||
|
|
PRE(user_max_width != 0);
|
||
|
|
|
||
|
|
// widths.size() is a proxy for the number of columns of the table.
|
||
|
|
const std::size_t total_padding = column_padding * (widths.size() - 1);
|
||
|
|
const text::widths_vector::value_type max_width = std::max(
|
||
|
|
user_max_width, total_padding) - total_padding;
|
||
|
|
|
||
|
|
const text::widths_vector::size_type refill_column =
|
||
|
|
find_refill_column(widths);
|
||
|
|
INV(refill_column < widths.size());
|
||
|
|
|
||
|
|
text::widths_vector::value_type width = 0;
|
||
|
|
for (text::widths_vector::size_type i = 0; i < widths.size(); ++i) {
|
||
|
|
if (i != refill_column)
|
||
|
|
width += widths[i];
|
||
|
|
}
|
||
|
|
widths[refill_column] = max_width - width;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Pads an input text to a specified width with spaces.
|
||
|
|
///
|
||
|
|
/// \param input The text to add padding to (may be empty).
|
||
|
|
/// \param length The desired length of the output.
|
||
|
|
/// \param is_last Whether the text being processed belongs to the last column
|
||
|
|
/// of a row or not. Values in the last column should not be padded to
|
||
|
|
/// prevent trailing whitespace on the screen (which affects copy/pasting
|
||
|
|
/// for example).
|
||
|
|
///
|
||
|
|
/// \return The padded cell. If the input string is longer than the desired
|
||
|
|
/// length, the input string is returned verbatim. The padded table won't be
|
||
|
|
/// correct, but we don't expect this to be a common case to worry about.
|
||
|
|
static std::string
|
||
|
|
pad_cell(const std::string& input, const std::size_t length, const bool is_last)
|
||
|
|
{
|
||
|
|
if (is_last)
|
||
|
|
return input;
|
||
|
|
else {
|
||
|
|
if (input.length() < length)
|
||
|
|
return input + std::string(length - input.length(), ' ');
|
||
|
|
else
|
||
|
|
return input;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Refills a cell and adds it to the output lines.
|
||
|
|
///
|
||
|
|
/// \param row The row containing the cell to be refilled.
|
||
|
|
/// \param widths The widths of the row.
|
||
|
|
/// \param column The column being refilled.
|
||
|
|
/// \param [in,out] textual_rows The output lines as processed so far. This is
|
||
|
|
/// updated to accomodate for the contents of the refilled cell, extending
|
||
|
|
/// the rows as necessary.
|
||
|
|
static void
|
||
|
|
refill_cell(const text::table_row& row, const text::widths_vector& widths,
|
||
|
|
const text::table_row::size_type column,
|
||
|
|
std::vector< text::table_row >& textual_rows)
|
||
|
|
{
|
||
|
|
const std::vector< std::string > rows = text::refill(row[column],
|
||
|
|
widths[column]);
|
||
|
|
|
||
|
|
if (textual_rows.size() < rows.size())
|
||
|
|
textual_rows.resize(rows.size(), text::table_row(row.size()));
|
||
|
|
|
||
|
|
for (std::vector< std::string >::size_type i = 0; i < rows.size(); ++i) {
|
||
|
|
for (text::table_row::size_type j = 0; j < row.size(); ++j) {
|
||
|
|
const bool is_last = j == row.size() - 1;
|
||
|
|
if (j == column)
|
||
|
|
textual_rows[i][j] = pad_cell(rows[i], widths[j], is_last);
|
||
|
|
else {
|
||
|
|
if (textual_rows[i][j].empty())
|
||
|
|
textual_rows[i][j] = pad_cell("", widths[j], is_last);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Formats a single table row.
|
||
|
|
///
|
||
|
|
/// \param row The row to format.
|
||
|
|
/// \param widths The widths of the columns to apply during formatting. Cells
|
||
|
|
/// wider than the specified width are refilled to attempt to fit in the
|
||
|
|
/// cell. Cells narrower than the width are right-padded with spaces.
|
||
|
|
/// \param separator The column separator to use.
|
||
|
|
///
|
||
|
|
/// \return The textual lines that contain the formatted row.
|
||
|
|
static std::vector< std::string >
|
||
|
|
format_row(const text::table_row& row, const text::widths_vector& widths,
|
||
|
|
const std::string& separator)
|
||
|
|
{
|
||
|
|
PRE(row.size() == widths.size());
|
||
|
|
|
||
|
|
std::vector< text::table_row > textual_rows(1, text::table_row(row.size()));
|
||
|
|
|
||
|
|
for (text::table_row::size_type column = 0; column < row.size(); ++column) {
|
||
|
|
if (widths[column] > row[column].length())
|
||
|
|
textual_rows[0][column] = pad_cell(row[column], widths[column],
|
||
|
|
column == row.size() - 1);
|
||
|
|
else
|
||
|
|
refill_cell(row, widths, column, textual_rows);
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector< std::string > lines;
|
||
|
|
for (std::vector< text::table_row >::const_iterator
|
||
|
|
iter = textual_rows.begin(); iter != textual_rows.end(); ++iter) {
|
||
|
|
lines.push_back(text::join(*iter, separator));
|
||
|
|
}
|
||
|
|
return lines;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
} // anonymous namespace
|
||
|
|
|
||
|
|
|
||
|
|
/// Constructs a new table.
|
||
|
|
///
|
||
|
|
/// \param ncolumns_ The number of columns that the table will have.
|
||
|
|
text::table::table(const table_row::size_type ncolumns_)
|
||
|
|
{
|
||
|
|
_column_widths.resize(ncolumns_, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Gets the number of columns in the table.
|
||
|
|
///
|
||
|
|
/// \return The number of columns in the table. This value remains constant
|
||
|
|
/// during the existence of the table.
|
||
|
|
text::widths_vector::size_type
|
||
|
|
text::table::ncolumns(void) const
|
||
|
|
{
|
||
|
|
return _column_widths.size();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Gets the width of a column.
|
||
|
|
///
|
||
|
|
/// The returned value is not valid if add_row() is called again, as the column
|
||
|
|
/// may have grown in width.
|
||
|
|
///
|
||
|
|
/// \param column The index of the column of which to get the width. Must be
|
||
|
|
/// less than the total number of columns.
|
||
|
|
///
|
||
|
|
/// \return The width of a column.
|
||
|
|
text::widths_vector::value_type
|
||
|
|
text::table::column_width(const widths_vector::size_type column) const
|
||
|
|
{
|
||
|
|
PRE(column < _column_widths.size());
|
||
|
|
return _column_widths[column];
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Gets the widths of all columns.
|
||
|
|
///
|
||
|
|
/// The returned value is not valid if add_row() is called again, as the columns
|
||
|
|
/// may have grown in width.
|
||
|
|
///
|
||
|
|
/// \return A vector with the width of all columns.
|
||
|
|
const text::widths_vector&
|
||
|
|
text::table::column_widths(void) const
|
||
|
|
{
|
||
|
|
return _column_widths;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Checks whether the table is empty or not.
|
||
|
|
///
|
||
|
|
/// \return True if the table is empty; false otherwise.
|
||
|
|
bool
|
||
|
|
text::table::empty(void) const
|
||
|
|
{
|
||
|
|
return _rows.empty();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Adds a row to the table.
|
||
|
|
///
|
||
|
|
/// \param row The row to be added. This row must have the same amount of
|
||
|
|
/// columns as defined during the construction of the table.
|
||
|
|
void
|
||
|
|
text::table::add_row(const table_row& row)
|
||
|
|
{
|
||
|
|
PRE(row.size() == _column_widths.size());
|
||
|
|
_rows.push_back(row);
|
||
|
|
|
||
|
|
for (table_row::size_type i = 0; i < row.size(); ++i)
|
||
|
|
if (_column_widths[i] < row[i].length())
|
||
|
|
_column_widths[i] = row[i].length();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Gets an iterator pointing to the beginning of the rows of the table.
|
||
|
|
///
|
||
|
|
/// \return An iterator on the rows.
|
||
|
|
text::table::const_iterator
|
||
|
|
text::table::begin(void) const
|
||
|
|
{
|
||
|
|
return _rows.begin();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Gets an iterator pointing to the end of the rows of the table.
|
||
|
|
///
|
||
|
|
/// \return An iterator on the rows.
|
||
|
|
text::table::const_iterator
|
||
|
|
text::table::end(void) const
|
||
|
|
{
|
||
|
|
return _rows.end();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Column width to denote that the column has to fit all of its cells.
|
||
|
|
const std::size_t text::table_formatter::width_auto = 0;
|
||
|
|
|
||
|
|
|
||
|
|
/// Column width to denote that the column can be refilled to fit the table.
|
||
|
|
const std::size_t text::table_formatter::width_refill =
|
||
|
|
std::numeric_limits< std::size_t >::max();
|
||
|
|
|
||
|
|
|
||
|
|
/// Constructs a new table formatter.
|
||
|
|
text::table_formatter::table_formatter(void) :
|
||
|
|
_separator(""),
|
||
|
|
_table_width(0)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Sets the width of a column.
|
||
|
|
///
|
||
|
|
/// All columns except one must have a width that is, at least, as wide as the
|
||
|
|
/// widest cell in the column. One of the columns can have a width of
|
||
|
|
/// width_refill, which indicates that the column will be refilled if the table
|
||
|
|
/// does not fit in its maximum width.
|
||
|
|
///
|
||
|
|
/// \param column The index of the column to set the width for.
|
||
|
|
/// \param width The width to set the column to.
|
||
|
|
///
|
||
|
|
/// \return A reference to this formatter to allow using the builder pattern.
|
||
|
|
text::table_formatter&
|
||
|
|
text::table_formatter::set_column_width(const table_row::size_type column,
|
||
|
|
const std::size_t width)
|
||
|
|
{
|
||
|
|
#if !defined(NDEBUG)
|
||
|
|
if (width == width_refill) {
|
||
|
|
for (widths_vector::size_type i = 0; i < _column_widths.size(); i++) {
|
||
|
|
if (i != column)
|
||
|
|
PRE_MSG(_column_widths[i] != width_refill,
|
||
|
|
"Only one column width can be set to width_refill");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
if (_column_widths.size() < column + 1)
|
||
|
|
_column_widths.resize(column + 1, width_auto);
|
||
|
|
_column_widths[column] = width;
|
||
|
|
return *this;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Sets the separator to use between the cells.
|
||
|
|
///
|
||
|
|
/// \param separator The separator to use.
|
||
|
|
///
|
||
|
|
/// \return A reference to this formatter to allow using the builder pattern.
|
||
|
|
text::table_formatter&
|
||
|
|
text::table_formatter::set_separator(const char* separator)
|
||
|
|
{
|
||
|
|
_separator = separator;
|
||
|
|
return *this;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Sets the maximum width of the table.
|
||
|
|
///
|
||
|
|
/// \param table_width The maximum width of the table; cannot be zero.
|
||
|
|
///
|
||
|
|
/// \return A reference to this formatter to allow using the builder pattern.
|
||
|
|
text::table_formatter&
|
||
|
|
text::table_formatter::set_table_width(const std::size_t table_width)
|
||
|
|
{
|
||
|
|
PRE(table_width > 0);
|
||
|
|
_table_width = table_width;
|
||
|
|
return *this;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/// Formats a table into a collection of textual lines.
|
||
|
|
///
|
||
|
|
/// \param t Table to format.
|
||
|
|
///
|
||
|
|
/// \return A collection of textual lines.
|
||
|
|
std::vector< std::string >
|
||
|
|
text::table_formatter::format(const table& t) const
|
||
|
|
{
|
||
|
|
std::vector< std::string > lines;
|
||
|
|
|
||
|
|
if (!t.empty()) {
|
||
|
|
widths_vector widths = override_column_widths(t, _column_widths);
|
||
|
|
if (_table_width != 0)
|
||
|
|
refill_widths(widths, _table_width, _separator.length());
|
||
|
|
|
||
|
|
for (table::const_iterator iter = t.begin(); iter != t.end(); ++iter) {
|
||
|
|
const std::vector< std::string > sublines =
|
||
|
|
format_row(*iter, widths, _separator);
|
||
|
|
std::copy(sublines.begin(), sublines.end(),
|
||
|
|
std::back_inserter(lines));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return lines;
|
||
|
|
}
|