mirror of
https://github.com/tdulcet/Table-and-Graph-Libs.git
synced 2025-05-05 21:41:12 +08:00
526 lines
13 KiB
C++
526 lines
13 KiB
C++
// Teal Dulcet, CS546
|
|
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <cstring>
|
|
#include <iomanip>
|
|
#include <cwchar>
|
|
#include <clocale>
|
|
#include <cstdlib>
|
|
#include <vector>
|
|
#include <iterator>
|
|
#include <sys/ioctl.h>
|
|
#include <unistd.h>
|
|
#include <regex>
|
|
|
|
namespace tables
|
|
{
|
|
using namespace std;
|
|
|
|
const char *const styles[][11] = {
|
|
{"-", "|", "+", "+", "+", "+", "+", "+", "+", "+", "+"}, // ASCII
|
|
{"—", "|", "+", "+", "+", "+", "+", "+", "+", "+", "+"}, // Basic
|
|
{"─", "│", "┌", "┬", "┐", "├", "┼", "┤", "└", "┴", "┘"}, // Light
|
|
{"━", "┃", "┏", "┳", "┓", "┣", "╋", "┫", "┗", "┻", "┛"}, // Heavy
|
|
{"═", "║", "╔", "╦", "╗", "╠", "╬", "╣", "╚", "╩", "╝"}, // Double
|
|
{"╌", "╎", "┌", "┬", "┐", "├", "┼", "┤", "└", "┴", "┘"}, // Light Dashed
|
|
{"╍", "╏", "┏", "┳", "┓", "┣", "╋", "┫", "┗", "┻", "┛"} // Heavy Dashed
|
|
};
|
|
// {" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " "}};//No border
|
|
|
|
regex ansi(R"(\x1B\[(?:[0-9]+(?:;[0-9]+)*)?m)");
|
|
|
|
struct options
|
|
{
|
|
bool headerrow = false;
|
|
bool headercolumn = false;
|
|
bool tableborder = true;
|
|
bool cellborder = false;
|
|
unsigned int padding = 1;
|
|
ios_base &(*alignment)(ios_base &) = left;
|
|
bool boolalpha = false;
|
|
const char *title = nullptr;
|
|
unsigned int style = 2;
|
|
};
|
|
|
|
const options defaultoptions;
|
|
|
|
// Number of columns needed to represent the string
|
|
// Adapted from: https://stackoverflow.com/a/31124065
|
|
int strcol(const char *str)
|
|
{
|
|
const string astr = regex_replace(str, ansi, "");
|
|
str = astr.c_str();
|
|
|
|
size_t length = strlen(str);
|
|
for (size_t i = 0; i < length; ++i)
|
|
if (iscntrl(str[i]))
|
|
{
|
|
cerr << "\nError! Control character in string.\n";
|
|
cout << "Control character: " << (int)str[i] << "\n";
|
|
}
|
|
|
|
length = mbstowcs(nullptr, str, 0);
|
|
if (length == static_cast<size_t>(-1))
|
|
{
|
|
cerr << "\nError! mbstowcs failed. Invalid multibyte character.\n";
|
|
exit(1);
|
|
}
|
|
++length;
|
|
|
|
wchar_t *wcstring = new wchar_t[length];
|
|
|
|
if (mbstowcs(wcstring, str, length) == static_cast<size_t>(-1))
|
|
{
|
|
if (wcstring != nullptr)
|
|
delete[] wcstring;
|
|
|
|
cerr << "\nError! mbstowcs failed. Invalid multibyte character.\n";
|
|
exit(1);
|
|
}
|
|
|
|
int width = wcswidth(wcstring, length);
|
|
if (width == -1)
|
|
{
|
|
cerr << "\nError! wcswidth failed. Nonprintable wide character.\n";
|
|
exit(1);
|
|
}
|
|
|
|
if (wcstring != nullptr)
|
|
delete[] wcstring;
|
|
|
|
return width;
|
|
}
|
|
|
|
// Word wrap
|
|
// Source: https://gist.github.com/tdulcet/819821ca69501822ad3f84a060c640a0
|
|
// Adapted from: https://stackoverflow.com/a/42016346 and https://stackoverflow.com/a/13094734
|
|
string wrap(const char *const str, const size_t line_length)
|
|
{
|
|
char words[strlen(str) + 1];
|
|
strcpy(words, str);
|
|
string wrapped;
|
|
|
|
size_t index = 0;
|
|
size_t linelen = 0;
|
|
while (words[index] != '\0')
|
|
{
|
|
if (words[index] == '\n')
|
|
{
|
|
linelen = 0;
|
|
}
|
|
else if (isspace(words[index]))
|
|
{
|
|
size_t tempindex = index + 1;
|
|
size_t templinelen = linelen;
|
|
while (!isspace(words[tempindex]) and words[tempindex] != '\0')
|
|
{
|
|
++templinelen;
|
|
|
|
++tempindex;
|
|
}
|
|
|
|
char temp[templinelen + 1];
|
|
strncpy(temp, words + (index - linelen), templinelen);
|
|
temp[templinelen] = '\0';
|
|
|
|
size_t width = strcol(temp);
|
|
|
|
if (width >= line_length)
|
|
{
|
|
words[index] = '\n';
|
|
linelen = 0;
|
|
}
|
|
}
|
|
|
|
if (words[index] == '\t')
|
|
linelen += 8 - (linelen % 8);
|
|
else if (words[index] != '\n')
|
|
++linelen;
|
|
|
|
++index;
|
|
}
|
|
wrapped = words;
|
|
return wrapped;
|
|
}
|
|
|
|
// Output char array as table
|
|
template <typename T>
|
|
int table(const vector<vector<basic_string<T>>> &array, const options &aoptions)
|
|
{
|
|
if (!size(array))
|
|
return 1;
|
|
|
|
const bool headerrow = aoptions.headerrow;
|
|
const bool headercolumn = aoptions.headercolumn;
|
|
const bool tableborder = aoptions.tableborder;
|
|
const bool cellborder = aoptions.cellborder;
|
|
const unsigned int padding = aoptions.padding;
|
|
const char *const title = aoptions.title;
|
|
const unsigned int style = aoptions.style;
|
|
|
|
if (style >= size(styles))
|
|
return 1;
|
|
|
|
const size_t rows = array.size();
|
|
const size_t columns = array[0].size();
|
|
|
|
int columnwidth[columns];
|
|
for (unsigned int j = 0; j < columns; ++j)
|
|
columnwidth[j] = 0;
|
|
|
|
int width = 0;
|
|
|
|
setlocale(LC_CTYPE, "");
|
|
|
|
for (unsigned int j = 0; j < columns; ++j)
|
|
{
|
|
for (unsigned int i = 0; i < rows; ++i)
|
|
{
|
|
int cellwidth = strcol(array[i][j].c_str());
|
|
if (cellwidth > columnwidth[j])
|
|
columnwidth[j] = cellwidth;
|
|
}
|
|
|
|
width += columnwidth[j];
|
|
}
|
|
|
|
struct winsize w;
|
|
ioctl(STDOUT_FILENO, TIOCGWINSZ, &w);
|
|
|
|
if (tableborder or cellborder or headerrow or headercolumn)
|
|
width += (((2 * padding) + 1) * columns) + (tableborder ? 1 : -1);
|
|
else
|
|
width += (2 * padding) * columns;
|
|
|
|
if (width > w.ws_col)
|
|
{
|
|
cerr << "The width of the table (" << width << ") is greater then the width of the terminal (" << w.ws_col << ").\n";
|
|
return 1;
|
|
}
|
|
|
|
if (title != nullptr and title[0] != '\0')
|
|
cout << wrap(title, w.ws_col) << "\n";
|
|
|
|
if (tableborder)
|
|
{
|
|
cout << styles[style][2];
|
|
|
|
for (unsigned int j = 0; j < columns; ++j)
|
|
{
|
|
for (unsigned int k = 0; k < (2 * padding) + columnwidth[j]; ++k)
|
|
cout << styles[style][0];
|
|
|
|
if (j == (columns - 1))
|
|
cout << styles[style][4] << "\n";
|
|
else if (cellborder or headerrow or (j == 0 and headercolumn))
|
|
cout << styles[style][3];
|
|
else
|
|
cout << styles[style][0];
|
|
}
|
|
}
|
|
|
|
for (unsigned int i = 0; i < rows; ++i)
|
|
{
|
|
for (unsigned int j = 0; j < columns; ++j)
|
|
{
|
|
if ((j == 0 and tableborder) or (j > 0 and cellborder) or (i == 0 and j > 0 and headerrow) or (j == 1 and headercolumn))
|
|
cout << styles[style][1];
|
|
else if (tableborder or (i > 0 and j > 0 and headerrow) or (j > 1 and headercolumn))
|
|
cout << " ";
|
|
|
|
const int difference = columnwidth[j] - strcol(array[i][j].c_str());
|
|
|
|
if ((i == 0 and headerrow) or (j == 0 and headercolumn))
|
|
{
|
|
const int apadding = (difference / 2);
|
|
|
|
cout << string(padding + apadding, ' ');
|
|
|
|
cout << "\e[1m" << array[i][j] << "\e[22m";
|
|
|
|
cout << string(padding + (difference - apadding), ' ');
|
|
}
|
|
else
|
|
{
|
|
cout << string(padding, ' ');
|
|
|
|
cout << aoptions.alignment << setw(difference + array[i][j].length()) << array[i][j];
|
|
|
|
cout << string(padding, ' ');
|
|
}
|
|
}
|
|
|
|
if (tableborder)
|
|
cout << styles[style][1];
|
|
|
|
cout << "\n";
|
|
|
|
if (tableborder)
|
|
{
|
|
if (i == (rows - 1))
|
|
cout << styles[style][8];
|
|
else if (cellborder or (i == 0 and headerrow) or headercolumn)
|
|
cout << styles[style][5];
|
|
}
|
|
|
|
if ((i == (rows - 1) and tableborder) or (i < (rows - 1) and cellborder) or (i == 0 and headerrow) or (i < (rows - 1) and headercolumn))
|
|
{
|
|
for (unsigned int j = 0; j < columns; ++j)
|
|
{
|
|
if ((i == (rows - 1) and tableborder) or (i < (rows - 1) and cellborder) or (i == 0 and headerrow) or (i < (rows - 1) and j == 0 and headercolumn))
|
|
for (unsigned int k = 0; k < (2 * padding) + columnwidth[j]; ++k)
|
|
cout << styles[style][0];
|
|
else if (i < (rows - 1) and headercolumn)
|
|
cout << string((2 * padding) + columnwidth[j], ' ');
|
|
|
|
if (j == (columns - 1))
|
|
{
|
|
if (tableborder)
|
|
{
|
|
if (i == (rows - 1))
|
|
cout << styles[style][10];
|
|
else if (cellborder or (i == 0 and headerrow))
|
|
cout << styles[style][7];
|
|
else if (headercolumn)
|
|
cout << styles[style][1];
|
|
}
|
|
|
|
cout << "\n";
|
|
}
|
|
else
|
|
{
|
|
if (i == (rows - 1) and tableborder)
|
|
{
|
|
if (cellborder or (j == 0 and headercolumn))
|
|
cout << styles[style][9];
|
|
else
|
|
cout << styles[style][0];
|
|
}
|
|
else if ((i < (rows - 1) and cellborder) or ((i == 0 and headerrow) and (j == 0 and headercolumn)))
|
|
cout << styles[style][6];
|
|
else if (i == 0 and headerrow)
|
|
cout << styles[style][9];
|
|
else if (i < (rows - 1) and headercolumn)
|
|
{
|
|
if (j == 0)
|
|
cout << styles[style][7];
|
|
else
|
|
cout << " ";
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
cout << endl;
|
|
|
|
return 0;
|
|
}
|
|
|
|
// Convert array to char array and output as table
|
|
template <typename T1, typename T2>
|
|
int array(const T1 &aarray, T2 headerrow[] = nullptr, T2 headercolumn[] = nullptr, const options &aoptions = defaultoptions)
|
|
{
|
|
if (!size(aarray))
|
|
return 1;
|
|
|
|
unsigned int i = 0;
|
|
unsigned int j = 0;
|
|
|
|
size_t rows = size(aarray);
|
|
size_t columns = size(aarray[0]);
|
|
|
|
if (!all_of(begin(aarray), end(aarray), [columns](auto &x)
|
|
{ return size(x) == columns; }))
|
|
{
|
|
cerr << "Error: The rows of the array must have the same number of columns.";
|
|
return 1;
|
|
}
|
|
|
|
if (headerrow != nullptr)
|
|
++rows;
|
|
|
|
if (headercolumn != nullptr)
|
|
++columns;
|
|
|
|
vector<vector<string>> aaarray(rows, vector<string>(columns));
|
|
|
|
if (headerrow != nullptr)
|
|
{
|
|
for (unsigned int j = 0; j < columns; ++j)
|
|
{
|
|
aaarray[i][j] = headerrow[j];
|
|
}
|
|
|
|
++i;
|
|
}
|
|
|
|
for (unsigned int ii = 0; i < rows; ++i)
|
|
{
|
|
if (headercolumn != nullptr)
|
|
{
|
|
unsigned int ii = i;
|
|
|
|
if (headerrow != nullptr)
|
|
--ii;
|
|
|
|
aaarray[i][j] = headercolumn[ii];
|
|
|
|
++j;
|
|
}
|
|
|
|
for (unsigned int jj = 0; j < columns; ++j)
|
|
{
|
|
ostringstream strm;
|
|
|
|
if (aoptions.boolalpha)
|
|
strm << boolalpha;
|
|
|
|
strm << aarray[ii][jj];
|
|
aaarray[i][j] = strm.str();
|
|
|
|
++jj;
|
|
}
|
|
|
|
j = 0;
|
|
++ii;
|
|
}
|
|
|
|
return table(aaarray, aoptions);
|
|
}
|
|
|
|
template <typename T>
|
|
int array(const size_t rows, const size_t columns, T **aarray, const char *const headerrow[] = nullptr, const char *const headercolumn[] = nullptr, const options &aoptions = defaultoptions)
|
|
{
|
|
vector<vector<T>> aaarray(rows, vector<T>(columns));
|
|
for (unsigned int i = 0; i < rows; ++i)
|
|
copy(aarray[i], aarray[i] + columns, aaarray[i].begin());
|
|
|
|
string *aheaderrow = nullptr;
|
|
string *aheadercolumn = nullptr;
|
|
|
|
if (headerrow and headercolumn)
|
|
{
|
|
vector<string> aaheaderrow(rows + 1);
|
|
copy(headerrow, headerrow + rows + 1, aaheaderrow.begin());
|
|
aheaderrow = aaheaderrow.data();
|
|
|
|
vector<string> aaheadercolumn(columns);
|
|
copy(headercolumn, headercolumn + columns, aaheadercolumn.begin());
|
|
aheadercolumn = aaheadercolumn.data();
|
|
}
|
|
else if (headerrow)
|
|
{
|
|
vector<string> aaheaderrow(rows);
|
|
copy(headerrow, headerrow + rows, aaheaderrow.begin());
|
|
aheaderrow = aaheaderrow.data();
|
|
}
|
|
else if (headercolumn)
|
|
{
|
|
vector<string> aaheadercolumn(columns);
|
|
copy(headercolumn, headercolumn + columns, aaheadercolumn.begin());
|
|
aheadercolumn = aaheadercolumn.data();
|
|
}
|
|
|
|
return array(aaarray, aheaderrow, aheadercolumn, aoptions);
|
|
}
|
|
|
|
// Convert one or more functions to array and output as table
|
|
template <typename T>
|
|
int functions(const long double xmin, const long double xmax, const long double xscl, const size_t numfunctions, function<T(T)> functions[], const options &aoptions = defaultoptions)
|
|
{
|
|
if (numfunctions == 0)
|
|
return 1;
|
|
|
|
if (xmin >= xmax)
|
|
{
|
|
cerr << "xmin must be less than xmax.\n";
|
|
return 1;
|
|
}
|
|
|
|
if (xscl <= 0)
|
|
{
|
|
cerr << "xscl must be greater than zero.\n";
|
|
return 1;
|
|
}
|
|
|
|
const size_t rows = ((xmax - xmin) * xscl) + 1;
|
|
const size_t columns = numfunctions + 1;
|
|
|
|
const char *const aheaderrow[] = {"x", "y"};
|
|
// const char* const aheaderrow[] = {"", "x", "y"};
|
|
|
|
const size_t length = size(aheaderrow);
|
|
|
|
string *headerrow = new string[columns];
|
|
|
|
for (unsigned int j = 0; j < columns; ++j)
|
|
{
|
|
if (j < (length - 1) or numfunctions == 1)
|
|
{
|
|
headerrow[j] = aheaderrow[j];
|
|
}
|
|
else
|
|
{
|
|
ostringstream strm;
|
|
strm << aheaderrow[length - 1] << j - length + 2;
|
|
headerrow[j] = strm.str();
|
|
}
|
|
}
|
|
|
|
string *headercolumn = nullptr;
|
|
// headercolumn = new string[rows + 1];
|
|
|
|
// for (unsigned int i = 0; i < rows + 1; ++i)
|
|
// {
|
|
// ostringstream strm;
|
|
// strm << i + 1;
|
|
// headercolumn[i] = strm.str();
|
|
// }
|
|
|
|
vector<vector<T>> aarray(rows, vector<T>(columns));
|
|
|
|
for (unsigned int i = 0; i < rows; ++i)
|
|
{
|
|
aarray[i][0] = (i / xscl) + xmin;
|
|
|
|
for (unsigned int j = 0; j < numfunctions; ++j)
|
|
aarray[i][j + 1] = (functions[j])(aarray[i][0]);
|
|
}
|
|
|
|
int code = array(aarray, headerrow, headercolumn, aoptions);
|
|
|
|
if (headerrow != nullptr)
|
|
{
|
|
delete[] headerrow;
|
|
}
|
|
|
|
// if (headercolumn != nullptr)
|
|
// {
|
|
// delete[] headercolumn;
|
|
// }
|
|
|
|
return code;
|
|
}
|
|
|
|
// Convert single function to array and output as table
|
|
template <typename T>
|
|
int function(const long double xmin, const long double xmax, const long double xscl, const function<T(T)> &afunction, const options &aoptions = defaultoptions)
|
|
{
|
|
std::function<T(T)> afunctions[] = {afunction};
|
|
|
|
return functions(xmin, xmax, xscl, 1, afunctions, aoptions);
|
|
}
|
|
|
|
// Convert single function to array and output as table
|
|
template <typename T>
|
|
int function(const long double xmin, const long double xmax, const long double xscl, T afunction(T), const options &aoptions = defaultoptions)
|
|
{
|
|
std::function<T(T)> afunctions[] = {afunction};
|
|
|
|
return functions(xmin, xmax, xscl, 1, afunctions, aoptions);
|
|
}
|
|
|
|
}
|