mirror of
https://github.com/tdulcet/Table-and-Graph-Libs.git
synced 2025-05-05 21:41:12 +08:00
340 lines
8.4 KiB
Python
340 lines
8.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# Teal Dulcet, CS546
|
|
|
|
import locale
|
|
import re
|
|
import shutil
|
|
import sys
|
|
import textwrap
|
|
from enum import IntEnum, auto
|
|
from typing import Any, Callable, List, Optional, Sequence, TextIO
|
|
|
|
if sys.platform != "win32":
|
|
import ctypes
|
|
from ctypes.util import find_library
|
|
|
|
libc = ctypes.CDLL(find_library("c"))
|
|
libc.wcwidth.argtypes = (ctypes.c_wchar,)
|
|
libc.wcwidth.restype = ctypes.c_int
|
|
libc.wcswidth.argtypes = (ctypes.c_wchar_p, ctypes.c_int)
|
|
libc.wcswidth.restype = ctypes.c_int
|
|
|
|
def wcswidth(astr: str) -> int:
|
|
return libc.wcswidth(astr, len(astr))
|
|
|
|
else:
|
|
from wcwidth import wcswidth
|
|
|
|
locale.setlocale(locale.LC_ALL, "")
|
|
|
|
|
|
class style_types(IntEnum):
|
|
ASCII = 0
|
|
basic = auto()
|
|
light = auto()
|
|
heavy = auto()
|
|
double = auto()
|
|
arc = auto()
|
|
light_dashed = auto()
|
|
heavy_dashed = auto()
|
|
|
|
|
|
styles = (
|
|
("-", "|", "+", "+", "+", "+", "+", "+", "+", "+", "+"), # ASCII
|
|
("—", "|", "+", "+", "+", "+", "+", "+", "+", "+", "+"), # Basic
|
|
("─", "│", "┌", "┬", "┐", "├", "┼", "┤", "└", "┴", "┘"), # Light
|
|
("━", "┃", "┏", "┳", "┓", "┣", "╋", "┫", "┗", "┻", "┛"), # Heavy
|
|
("═", "║", "╔", "╦", "╗", "╠", "╬", "╣", "╚", "╩", "╝"), # Double
|
|
("─", "│", "╭", "┬", "╮", "├", "┼", "┤", "╰", "┴", "╯"), # Light Arc
|
|
("╌", "┊", "┌", "┬", "┐", "├", "┼", "┤", "└", "┴", "┘"), # Light Dashed
|
|
("╍", "┋", "┏", "┳", "┓", "┣", "╋", "┫", "┗", "┻", "┛"), # Heavy Dashed
|
|
# (" ", " ", " ", " ", " ", " ", " ", " ", " ", " ", " ")) #No border
|
|
)
|
|
|
|
ansi = re.compile(r"\x1B\[(?:[0-9]+(?:;[0-9]+)*)?m")
|
|
|
|
|
|
def strcol(astr: str) -> int:
|
|
"""Returns the number of columns that the given string would take up if printed."""
|
|
astr = ansi.sub("", astr)
|
|
width = wcswidth(astr)
|
|
if width == -1:
|
|
msg = "wcswidth failed. Nonprintable wide character."
|
|
raise ValueError(msg)
|
|
return width
|
|
# return len(astr)
|
|
|
|
|
|
def table(
|
|
array: List[List[str]],
|
|
headerrow: bool = False,
|
|
headercolumn: bool = False,
|
|
tableborder: bool = True,
|
|
cellborder: bool = False,
|
|
padding: int = 1,
|
|
alignment: str = "",
|
|
title: Optional[str] = None,
|
|
style: style_types = style_types.light,
|
|
file: TextIO = sys.stdout,
|
|
check: bool = True,
|
|
) -> int:
|
|
"""Output char array as table."""
|
|
if not array:
|
|
return 1
|
|
|
|
rows = len(array)
|
|
columns = len(array[0])
|
|
|
|
columnwidth = [max(strcol(i) for i in j) for j in zip(*array)]
|
|
|
|
width = sum(columnwidth)
|
|
|
|
w = shutil.get_terminal_size()
|
|
|
|
if tableborder or cellborder or headerrow or headercolumn:
|
|
width += (2 * padding + 1) * columns + (1 if tableborder else -1)
|
|
else:
|
|
width += 2 * padding * columns
|
|
|
|
if check and width > w.columns:
|
|
print(f"The width of the table ({width}) is greater then the width of the terminal ({w.columns}).", file=sys.stderr)
|
|
return 1
|
|
|
|
if title:
|
|
print(textwrap.fill(title, width=width), file=file)
|
|
|
|
astyle = styles[style]
|
|
|
|
strm = ""
|
|
|
|
if tableborder:
|
|
strm += astyle[2]
|
|
|
|
for j in range(columns):
|
|
strm += astyle[0] * (2 * padding + columnwidth[j])
|
|
|
|
if j < columns - 1:
|
|
if cellborder or headerrow or (not j and headercolumn):
|
|
strm += astyle[3]
|
|
else:
|
|
strm += astyle[0]
|
|
|
|
strm += astyle[4] + "\n"
|
|
|
|
for i in range(rows):
|
|
if tableborder:
|
|
strm += astyle[1]
|
|
|
|
for j in range(columns):
|
|
if (j > 0 and cellborder) or (not i and j > 0 and headerrow) or (j == 1 and headercolumn):
|
|
strm += astyle[1]
|
|
elif j > 0 and (tableborder or (i > 0 and headerrow) or headercolumn):
|
|
strm += " "
|
|
|
|
awidth = columnwidth[j] - (strcol(array[i][j]) - len(array[i][j]))
|
|
|
|
if (not i and headerrow) or (not j and headercolumn):
|
|
strm += (" " * padding) + "\033[1m" + f"{array[i][j]:^{awidth}}" + "\033[22m" + (" " * padding)
|
|
else:
|
|
strm += (" " * padding) + f"{array[i][j]:{alignment}{awidth}}" + (" " * padding)
|
|
|
|
if tableborder:
|
|
strm += astyle[1]
|
|
|
|
if i < rows - 1 or tableborder:
|
|
strm += "\n"
|
|
|
|
if (i < rows - 1 and cellborder) or (not i and headerrow) or (i < rows - 1 and headercolumn):
|
|
if tableborder and (cellborder or (not i and headerrow) or headercolumn):
|
|
strm += astyle[5]
|
|
|
|
for j in range(columns):
|
|
if cellborder or (not i and headerrow) or (not j and headercolumn):
|
|
strm += astyle[0] * (2 * padding + columnwidth[j])
|
|
elif headercolumn:
|
|
strm += " " * (2 * padding + columnwidth[j])
|
|
|
|
if j < columns - 1:
|
|
if cellborder or ((not i and headerrow) and (not j and headercolumn)):
|
|
strm += astyle[6]
|
|
elif not i and headerrow:
|
|
strm += astyle[9]
|
|
elif headercolumn:
|
|
if not j:
|
|
strm += astyle[7]
|
|
else:
|
|
strm += " "
|
|
|
|
if tableborder:
|
|
if cellborder or (not i and headerrow):
|
|
strm += astyle[7]
|
|
elif headercolumn:
|
|
strm += astyle[1]
|
|
|
|
strm += "\n"
|
|
|
|
if tableborder:
|
|
strm += astyle[8]
|
|
|
|
for j in range(columns):
|
|
strm += astyle[0] * (2 * padding + columnwidth[j])
|
|
|
|
if j < columns - 1:
|
|
if cellborder or (not j and headercolumn):
|
|
strm += astyle[9]
|
|
else:
|
|
strm += astyle[0]
|
|
|
|
strm += astyle[10]
|
|
|
|
print(strm, file=file)
|
|
|
|
return 0
|
|
|
|
|
|
def array(
|
|
aarray: Sequence[Sequence[Any]],
|
|
aheaderrow: Optional[Sequence[str]] = None,
|
|
aheadercolumn: Optional[Sequence[str]] = None,
|
|
headerrow: bool = False,
|
|
headercolumn: bool = False,
|
|
tableborder: bool = True,
|
|
cellborder: bool = False,
|
|
padding: int = 1,
|
|
alignment: str = "",
|
|
title: Optional[str] = None,
|
|
style: style_types = style_types.light,
|
|
file: TextIO = sys.stdout,
|
|
check: bool = True,
|
|
) -> int:
|
|
"""Convert array to char array and output as table."""
|
|
if not aarray:
|
|
return 1
|
|
|
|
rows = len(aarray)
|
|
columns = len(aarray[0])
|
|
|
|
if not all(len(x) == columns for x in aarray):
|
|
print("Error: The rows of the array must have the same number of columns.", file=sys.stderr)
|
|
return 1
|
|
|
|
if aheaderrow:
|
|
rows += 1
|
|
|
|
if aheadercolumn:
|
|
columns += 1
|
|
|
|
if aheaderrow and len(aheaderrow) != columns:
|
|
print("Error: The header row must have the same number of columns as the array.", file=sys.stderr)
|
|
return 1
|
|
|
|
if aheadercolumn and len(aheadercolumn) != (rows - 1 if aheaderrow else rows):
|
|
print("Error: The header column must have the same number of rows as the array.", file=sys.stderr)
|
|
return 1
|
|
|
|
aaarray = [["" for j in range(columns)] for i in range(rows)]
|
|
|
|
if aheaderrow:
|
|
aaarray[0] = aheaderrow[:columns]
|
|
|
|
for i in range(1 if aheaderrow else 0, rows):
|
|
ii = i - 1 if aheaderrow else i
|
|
|
|
if aheadercolumn:
|
|
aaarray[i][0] = aheadercolumn[ii]
|
|
|
|
j = 1 if aheadercolumn else 0
|
|
aaarray[i][j:] = map(str, aarray[ii][: columns - j])
|
|
|
|
return table(aaarray, headerrow, headercolumn, tableborder, cellborder, padding, alignment, title, style, file, check)
|
|
|
|
|
|
def functions(
|
|
xmin: float,
|
|
xmax: float,
|
|
xstep: float,
|
|
afunctions: Sequence[Callable[[float], float]],
|
|
headerrow: bool = False,
|
|
headercolumn: bool = False,
|
|
tableborder: bool = True,
|
|
cellborder: bool = False,
|
|
padding: int = 1,
|
|
alignment: str = "",
|
|
title: Optional[str] = None,
|
|
style: style_types = style_types.light,
|
|
file: TextIO = sys.stdout,
|
|
check: bool = True,
|
|
) -> int:
|
|
"""Convert one or more functions to array and output as table."""
|
|
if not afunctions:
|
|
return 1
|
|
|
|
if xmin >= xmax:
|
|
print("xmin must be less than xmax.", file=sys.stderr)
|
|
return 1
|
|
|
|
if xstep <= 0:
|
|
print("xstep must be greater than zero.", file=sys.stderr)
|
|
return 1
|
|
|
|
rows = int((xmax - xmin) / xstep) + 1
|
|
columns = len(afunctions) + 1
|
|
|
|
aaheaderrow = ["x", "y"]
|
|
length = len(aaheaderrow)
|
|
|
|
aheaderrow = [""] * columns
|
|
|
|
aheaderrow = (
|
|
aaheaderrow
|
|
if len(afunctions) == 1
|
|
else aaheaderrow[:-1] + [aaheaderrow[-1] + str(j - length + 2) for j in range(1, columns)]
|
|
)
|
|
|
|
aarray = [[0 for j in range(columns)] for i in range(rows)]
|
|
|
|
for i in range(rows):
|
|
temp = aarray[i][0] = i * xstep + xmin
|
|
|
|
aarray[i][1:] = [function(temp) for function in afunctions]
|
|
|
|
return array(
|
|
aarray, aheaderrow, None, headerrow, headercolumn, tableborder, cellborder, padding, alignment, title, style, file, check
|
|
)
|
|
|
|
|
|
def function(
|
|
xmin: float,
|
|
xmax: float,
|
|
xstep: float,
|
|
afunction: Callable[[float], float],
|
|
headerrow: bool = False,
|
|
headercolumn: bool = False,
|
|
tableborder: bool = True,
|
|
cellborder: bool = False,
|
|
padding: int = 1,
|
|
alignment: str = "",
|
|
title: Optional[str] = None,
|
|
style: style_types = style_types.light,
|
|
file: TextIO = sys.stdout,
|
|
check: bool = True,
|
|
) -> int:
|
|
"""Convert single function to array and output as table."""
|
|
return functions(
|
|
xmin,
|
|
xmax,
|
|
xstep,
|
|
(afunction,),
|
|
headerrow,
|
|
headercolumn,
|
|
tableborder,
|
|
cellborder,
|
|
padding,
|
|
alignment,
|
|
title,
|
|
style,
|
|
file,
|
|
check,
|
|
)
|