mirror of
https://github.com/tdulcet/Table-and-Graph-Libs.git
synced 2025-05-06 06:35:26 +08:00
1024 lines
27 KiB
Python
1024 lines
27 KiB
Python
#!/usr/bin/env python3
|
||
|
||
# Teal Dulcet, CS546
|
||
|
||
import locale
|
||
import math
|
||
import shutil
|
||
import sys
|
||
import textwrap
|
||
from datetime import datetime, timezone
|
||
from enum import Enum, IntEnum, auto
|
||
from fractions import Fraction
|
||
from typing import Callable, List, Optional, Sequence, TextIO, Tuple
|
||
|
||
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
|
||
)
|
||
|
||
|
||
class color_types(IntEnum):
|
||
default = 0
|
||
black = auto()
|
||
red = auto()
|
||
green = auto()
|
||
yellow = auto()
|
||
blue = auto()
|
||
magenta = auto()
|
||
cyan = auto()
|
||
white = auto()
|
||
gray = auto()
|
||
bright_red = auto()
|
||
bright_green = auto()
|
||
bright_yellow = auto()
|
||
bright_blue = auto()
|
||
bright_magenta = auto()
|
||
bright_cyan = auto()
|
||
bright_white = auto()
|
||
|
||
|
||
colors = (39, 30, 31, 32, 33, 34, 35, 36, 37, 90, 91, 92, 93, 94, 95, 96, 97)
|
||
|
||
# fmt: off
|
||
dots = (
|
||
"⠀", "⠁", "⠂", "⠃", "⠄", "⠅", "⠆", "⠇", "⠈", "⠉", "⠊", "⠋", "⠌", "⠍", "⠎",
|
||
"⠏", "⠐", "⠑", "⠒", "⠓", "⠔", "⠕", "⠖", "⠗", "⠘", "⠙", "⠚", "⠛", "⠜", "⠝",
|
||
"⠞", "⠟", "⠠", "⠡", "⠢", "⠣", "⠤", "⠥", "⠦", "⠧", "⠨", "⠩", "⠪", "⠫", "⠬",
|
||
"⠭", "⠮", "⠯", "⠰", "⠱", "⠲", "⠳", "⠴", "⠵", "⠶", "⠷", "⠸", "⠹", "⠺", "⠻",
|
||
"⠼", "⠽", "⠾", "⠿", "⡀", "⡁", "⡂", "⡃", "⡄", "⡅", "⡆", "⡇", "⡈", "⡉", "⡊",
|
||
"⡋", "⡌", "⡍", "⡎", "⡏", "⡐", "⡑", "⡒", "⡓", "⡔", "⡕", "⡖", "⡗", "⡘", "⡙",
|
||
"⡚", "⡛", "⡜", "⡝", "⡞", "⡟", "⡠", "⡡", "⡢", "⡣", "⡤", "⡥", "⡦", "⡧", "⡨",
|
||
"⡩", "⡪", "⡫", "⡬", "⡭", "⡮", "⡯", "⡰", "⡱", "⡲", "⡳", "⡴", "⡵", "⡶", "⡷",
|
||
"⡸", "⡹", "⡺", "⡻", "⡼", "⡽", "⡾", "⡿", "⢀", "⢁", "⢂", "⢃", "⢄", "⢅", "⢆",
|
||
"⢇", "⢈", "⢉", "⢊", "⢋", "⢌", "⢍", "⢎", "⢏", "⢐", "⢑", "⢒", "⢓", "⢔", "⢕",
|
||
"⢖", "⢗", "⢘", "⢙", "⢚", "⢛", "⢜", "⢝", "⢞", "⢟", "⢠", "⢡", "⢢", "⢣", "⢤",
|
||
"⢥", "⢦", "⢧", "⢨", "⢩", "⢪", "⢫", "⢬", "⢭", "⢮", "⢯", "⢰", "⢱", "⢲", "⢳",
|
||
"⢴", "⢵", "⢶", "⢷", "⢸", "⢹", "⢺", "⢻", "⢼", "⢽", "⢾", "⢿", "⣀", "⣁", "⣂",
|
||
"⣃", "⣄", "⣅", "⣆", "⣇", "⣈", "⣉", "⣊", "⣋", "⣌", "⣍", "⣎", "⣏", "⣐", "⣑",
|
||
"⣒", "⣓", "⣔", "⣕", "⣖", "⣗", "⣘", "⣙", "⣚", "⣛", "⣜", "⣝", "⣞", "⣟", "⣠",
|
||
"⣡", "⣢", "⣣", "⣤", "⣥", "⣦", "⣧", "⣨", "⣩", "⣪", "⣫", "⣬", "⣭", "⣮", "⣯",
|
||
"⣰", "⣱", "⣲", "⣳", "⣴", "⣵", "⣶", "⣷", "⣸", "⣹", "⣺", "⣻", "⣼", "⣽", "⣾",
|
||
"⣿")
|
||
dotvalues = ((0x1, 0x2, 0x4, 0x40), (0x8, 0x10, 0x20, 0x80))
|
||
|
||
blocks = ("\xA0", "█")
|
||
|
||
blocks_quadrant = ("\xA0", "▘", "▝", "▀", "▖", "▌", "▞", "▛", "▗", "▚", "▐", "▜", "▄", "▙", "▟", "█")
|
||
|
||
separated_blocks_quadrant = ("\xA0", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "")
|
||
|
||
blocks_sextant = ("\xA0", "🬀", "🬁", "🬂", "🬃", "🬄", "🬅", "🬆", "🬇", "🬈", "🬉", "🬊", "🬋", "🬌", "🬍", "🬎", "🬏", "🬐", "🬑", "🬒", "🬓", "▌", "🬔", "🬕", "🬖", "🬗", "🬘", "🬙", "🬚", "🬛", "🬜", "🬝", "🬞", "🬟", "🬠", "🬡", "🬢", "🬣", "🬤", "🬥", "🬦", "🬧", "▐", "🬨", "🬩", "🬪", "🬫", "🬬", "🬭", "🬮", "🬯", "🬰", "🬱", "🬲", "🬳", "🬴", "🬵", "🬶", "🬷", "🬸", "🬹", "🬺", "🬻", "█")
|
||
|
||
separated_blocks_sextant = ("\xA0", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "")
|
||
|
||
blocks_octant = ("\xA0", "", "", "🮂", "", "▘", "", "", "", "", "▝", "", "", "", "", "▀", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "🮅", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "▖", "", "", "", "", "▌", "", "", "", "", "▞", "", "", "", "", "▛", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "▗", "", "", "", "", "▚", "", "", "", "", "▐", "", "", "", "", "▜", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "▂", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "▄", "", "", "", "", "▙", "", "", "", "", "▟", "", "▆", "", "", "█")
|
||
|
||
bars = ("\xA0", "▁", "▂", "▃", "▄", "▅", "▆", "▇", "█")
|
||
# fmt: on
|
||
|
||
|
||
class type_types(IntEnum):
|
||
braille = 0
|
||
block = auto()
|
||
block_quadrant = auto()
|
||
separated_block_quadrant = auto()
|
||
block_sextant = auto()
|
||
separated_block_sextant = auto()
|
||
block_octant = auto()
|
||
histogram = auto() # Set automatically by the histogram() function
|
||
|
||
|
||
atype_types = (
|
||
type_types.braille,
|
||
type_types.block,
|
||
type_types.block_quadrant,
|
||
type_types.separated_block_quadrant,
|
||
type_types.block_sextant,
|
||
type_types.separated_block_sextant,
|
||
type_types.block_octant,
|
||
)
|
||
|
||
densities = ((4, 2), (1, 1), (2, 2), (2, 2), (3, 2), (3, 2), (4, 2), (8, 1))
|
||
|
||
marks = (
|
||
((0, 0),),
|
||
((0, 1), (-1, 0), (0, 0), (1, 0), (0, -1)),
|
||
((-1, 1), (0, 1), (1, 1), (-1, 0), (1, 0), (-1, -1), (0, -1), (1, -1)),
|
||
)
|
||
|
||
|
||
class mark_types(IntEnum):
|
||
dot = 0
|
||
plus = auto()
|
||
square = auto()
|
||
|
||
|
||
fractions = {
|
||
"¼": Fraction(1, 4),
|
||
"½": Fraction(1, 2),
|
||
"¾": Fraction(3, 4),
|
||
"⅐": Fraction(1, 7),
|
||
"⅑": Fraction(1, 9),
|
||
"⅒": Fraction(1, 10),
|
||
"⅓": Fraction(1, 3),
|
||
"⅔": Fraction(2, 3),
|
||
"⅕": Fraction(1, 5),
|
||
"⅖": Fraction(2, 5),
|
||
"⅗": Fraction(3, 5),
|
||
"⅘": Fraction(4, 5),
|
||
"⅙": Fraction(1, 6),
|
||
"⅚": Fraction(5, 6),
|
||
"⅛": Fraction(1, 8),
|
||
"⅜": Fraction(3, 8),
|
||
"⅝": Fraction(5, 8),
|
||
"⅞": Fraction(7, 8),
|
||
}
|
||
|
||
constants = {"π": math.pi, "e": math.e}
|
||
|
||
|
||
class units_types(Enum):
|
||
number = auto()
|
||
scale_none = auto()
|
||
scale_SI = auto()
|
||
scale_IEC = auto()
|
||
scale_IEC_I = auto()
|
||
fracts = auto()
|
||
percent = auto()
|
||
date = auto()
|
||
time = auto()
|
||
monetary = auto()
|
||
|
||
|
||
suffix_power_char = ("", "K", "M", "G", "T", "P", "E", "Z", "Y", "R", "Q")
|
||
|
||
MAX = sys.float_info.radix**sys.float_info.mant_dig - 1
|
||
|
||
|
||
def strcol(astr: str) -> int:
|
||
"""Returns the number of columns that the given string would take up if printed."""
|
||
width = wcswidth(astr)
|
||
if width == -1:
|
||
msg = "wcswidth failed. Nonprintable wide character."
|
||
raise ValueError(msg)
|
||
return width
|
||
# return len(astr)
|
||
|
||
|
||
def outputunit(number: float, scale: units_types) -> str:
|
||
x = 0
|
||
val = number
|
||
if -sys.float_info.max <= val <= sys.float_info.max:
|
||
while abs(val) >= 10:
|
||
x += 1
|
||
val /= 10
|
||
|
||
if scale == units_types.scale_none:
|
||
if x > sys.float_info.dig:
|
||
return ""
|
||
|
||
return f"{number:.{sys.float_info.dig}n}"
|
||
|
||
if x > 33 - 1:
|
||
return ""
|
||
|
||
if scale in {units_types.scale_IEC, units_types.scale_IEC_I}:
|
||
scale_base = 1024
|
||
elif scale == units_types.scale_SI:
|
||
scale_base = 1000
|
||
|
||
power = 0
|
||
if -sys.float_info.max <= number <= sys.float_info.max:
|
||
while abs(number) >= scale_base:
|
||
power += 1
|
||
number /= scale_base
|
||
|
||
anumber = abs(number)
|
||
anumber += 0.0005 if anumber < 10 else 0.005 if anumber < 100 else 0.05 if anumber < 1000 else 0.5
|
||
|
||
if number and anumber < 1000 and power > 0:
|
||
strm = f"{number:.{sys.float_info.dig}n}"
|
||
|
||
length = 5 + (number < 0)
|
||
if len(strm) > length:
|
||
prec = 3 if anumber < 10 else 2 if anumber < 100 else 1
|
||
strm = locale.format_string(f"%.{prec}f", number, grouping=True)
|
||
else:
|
||
strm = locale.format_string("%.0f", number, grouping=True)
|
||
|
||
# "k" if power == 1 and scale == scale_SI else
|
||
strm += suffix_power_char[power] if power < len(suffix_power_char) else "(error)"
|
||
|
||
if scale == units_types.scale_IEC_I and power > 0:
|
||
strm += "i"
|
||
|
||
return strm
|
||
|
||
|
||
def outputfraction(number: float) -> str:
|
||
"""Convert fractions and constants to Unicode characters."""
|
||
output = False
|
||
|
||
strm = ""
|
||
|
||
n = abs(number)
|
||
|
||
if n <= MAX:
|
||
# fractionpart, intpart = math.modf(number)
|
||
# fractionpart = abs(fractionpart)
|
||
intpart, fractionpart = divmod(Fraction(number).limit_denominator(), 1)
|
||
|
||
for fraction, value in fractions.items():
|
||
if abs(fractionpart - value) <= sys.float_info.epsilon * n:
|
||
if intpart == 0 and number < 0:
|
||
strm += "-"
|
||
elif intpart != 0:
|
||
strm += f"{intpart:n}"
|
||
|
||
strm += fraction
|
||
|
||
output = True
|
||
break
|
||
|
||
if n > sys.float_info.epsilon and not output:
|
||
for constant, value in constants.items():
|
||
if not output and number % value <= sys.float_info.epsilon * n:
|
||
intpart = number / value
|
||
|
||
if intpart == -1:
|
||
strm += "-"
|
||
elif intpart != 1:
|
||
strm += f"{intpart:.{sys.float_info.dig}n}"
|
||
|
||
strm += constant
|
||
|
||
output = True
|
||
break
|
||
|
||
if not output:
|
||
strm += f"{number:n}"
|
||
|
||
return strm
|
||
|
||
|
||
def outputlabel(label: float, units: units_types) -> Tuple[int, str]:
|
||
"""Outputs a label in a nice, human readable format."""
|
||
if units == units_types.number:
|
||
strm = f"{label:n}"
|
||
elif units in {units_types.scale_none, units_types.scale_SI, units_types.scale_IEC, units_types.scale_IEC_I}:
|
||
strm = outputunit(label, units)
|
||
elif units == units_types.fracts:
|
||
strm = outputfraction(label)
|
||
elif units == units_types.percent:
|
||
strm = f"{label:%}"
|
||
elif units == units_types.date:
|
||
strm = datetime.fromtimestamp(label, timezone.utc).strftime("%x")
|
||
elif units == units_types.time:
|
||
strm = datetime.fromtimestamp(label, timezone.utc).strftime("%X")
|
||
elif units == units_types.monetary:
|
||
strm = locale.currency(label, grouping=True)
|
||
|
||
length = strcol(strm)
|
||
|
||
return length, strm
|
||
|
||
|
||
def outputcolor(color: color_types) -> str:
|
||
return f"\033[{colors[color]}m"
|
||
|
||
|
||
def graph(
|
||
height: int,
|
||
width: int,
|
||
xmin: float,
|
||
xmax: float,
|
||
ymin: float,
|
||
ymax: float,
|
||
array: List[List[int]],
|
||
border: bool = False,
|
||
axis: bool = True,
|
||
axislabel: bool = True,
|
||
axistick: bool = True,
|
||
axisunitslabel: bool = True,
|
||
xunits: units_types = units_types.fracts,
|
||
yunits: units_types = units_types.fracts,
|
||
atype: type_types = type_types.braille,
|
||
style: style_types = style_types.light,
|
||
title: Optional[str] = None,
|
||
file: TextIO = sys.stdout,
|
||
check: bool = True,
|
||
) -> int:
|
||
"""Output graph."""
|
||
if not array:
|
||
return 1
|
||
|
||
if not height:
|
||
return 1
|
||
|
||
if not width:
|
||
return 1
|
||
|
||
w = shutil.get_terminal_size()
|
||
|
||
ai, aj = densities[atype]
|
||
|
||
aheight = height // ai
|
||
awidth = width // aj
|
||
|
||
if check:
|
||
if aheight > w.lines:
|
||
print(f"The height of the graph ({aheight}) is greater then the height of the terminal ({w.lines}).", file=sys.stderr)
|
||
return 1
|
||
|
||
if awidth > w.columns:
|
||
print(f"The width of the graph ({awidth}) is greater then the width of the terminal ({w.columns}).", file=sys.stderr)
|
||
return 1
|
||
|
||
if xmin >= xmax:
|
||
print("xmin must be less than xmax.", file=sys.stderr)
|
||
return 1
|
||
|
||
if ymin >= ymax:
|
||
print("ymin must be less than ymax.", file=sys.stderr)
|
||
return 1
|
||
|
||
xstep = (xmax - xmin) / width
|
||
ystep = (ymax - ymin) / height
|
||
xaxis = 0 if xmin > 0 else width if xmax < 0 else width - xmax / xstep
|
||
yaxis = height if ymin > 0 else 0 if ymax < 0 else ymax / ystep
|
||
xdivisor = 4 * aj * ((((2 * width) // aj) // 160) + 2)
|
||
ydivisor = 2 * ai * ((((4 * height) // ai) // 160) + 2)
|
||
|
||
if title:
|
||
print(textwrap.fill(title, width=awidth), file=file)
|
||
|
||
astyle = styles[style]
|
||
|
||
strm = ""
|
||
|
||
if border:
|
||
strm += astyle[2] + (astyle[0] * awidth) + astyle[4] + "\n"
|
||
|
||
i = 0
|
||
while i < height:
|
||
ayaxis = i <= yaxis < i + ai if yaxis <= height - ai else i < yaxis <= i + ai
|
||
yaxislabel = i <= yaxis + ai < i + ai if yaxis <= height - ai else i < yaxis - ai <= i + ai
|
||
|
||
ylabelstrm = ""
|
||
ylabellength = 0
|
||
|
||
if axis and axistick and axisunitslabel and 0 <= yaxis <= height:
|
||
output = False
|
||
label = 0.0
|
||
adivisor = -ydivisor if i < yaxis else ydivisor
|
||
|
||
k = yaxis + adivisor
|
||
while (k >= i if i < yaxis else k < i + ai) and i >= ai and not output:
|
||
if i <= k < i + ai:
|
||
label = ymax - min(k, height) * ystep
|
||
|
||
output = True
|
||
k += adivisor
|
||
|
||
if output:
|
||
ylabellength, ylabelstrm = outputlabel(label, yunits)
|
||
ylabellength *= aj
|
||
|
||
if border:
|
||
strm += astyle[1]
|
||
|
||
j = 0
|
||
while j < width:
|
||
axaxis = j < xaxis <= j + aj if xaxis >= aj else j <= xaxis < j + aj
|
||
xaxislabel = j < xaxis - aj <= j + aj if xaxis >= aj else j <= xaxis + aj < j + aj
|
||
|
||
output = False
|
||
|
||
if axis:
|
||
if axaxis and ayaxis:
|
||
strm += astyle[6]
|
||
output = True
|
||
elif axaxis:
|
||
if not i:
|
||
strm += astyle[4]
|
||
output = True
|
||
elif i >= height - ai:
|
||
strm += astyle[10]
|
||
output = True
|
||
elif axistick:
|
||
adivisor = -ydivisor if i < yaxis else ydivisor
|
||
|
||
k = yaxis + adivisor
|
||
while (k >= i if i < yaxis else k < i + ai) and i >= ai and not output:
|
||
if i <= k < i + ai:
|
||
strm += astyle[7 if xaxis >= aj else 5]
|
||
output = True
|
||
k += adivisor
|
||
if not output:
|
||
strm += astyle[1]
|
||
output = True
|
||
elif ayaxis:
|
||
if not j:
|
||
strm += astyle[2]
|
||
output = True
|
||
elif j >= width - aj:
|
||
strm += astyle[4]
|
||
output = True
|
||
elif axistick:
|
||
adivisor = -xdivisor if j < xaxis else xdivisor
|
||
|
||
k = xaxis + adivisor
|
||
while (k >= j if j < xaxis else k < j + aj) and j < width - ai and not output:
|
||
if j <= k < j + aj:
|
||
strm += astyle[3 if yaxis <= height - ai else 9]
|
||
output = True
|
||
k += adivisor
|
||
if not output:
|
||
strm += astyle[0]
|
||
output = True
|
||
elif yaxislabel and xaxislabel and axistick and axisunitslabel and ymin <= 0 <= ymax and xmin <= 0 <= xmax:
|
||
strm += "0"
|
||
output = True
|
||
elif (j >= width - aj if xaxis <= width - aj else not j) and yaxislabel and axislabel:
|
||
strm += "x"
|
||
output = True
|
||
elif yaxislabel and axistick and axisunitslabel:
|
||
label = 0.0
|
||
adivisor = -xdivisor if j < xaxis else xdivisor
|
||
if j < xaxis:
|
||
j += aj
|
||
|
||
k = xaxis + adivisor
|
||
while (k >= j if j < xaxis else k < j + aj) and j < width - aj and not output:
|
||
if j <= k < j + aj:
|
||
label = min(k, width) * xstep + xmin
|
||
|
||
output = True
|
||
k += adivisor
|
||
|
||
if adivisor < 0:
|
||
j -= aj
|
||
|
||
if output:
|
||
output = False
|
||
|
||
length, astrm = outputlabel(label, xunits)
|
||
length *= aj
|
||
if (j >= xaxis or j + length < xaxis - ai) and j + length < width - aj:
|
||
strm += astrm
|
||
|
||
if length > aj:
|
||
j += length - aj
|
||
|
||
if adivisor < 0:
|
||
output = True
|
||
else:
|
||
j += aj
|
||
elif (not i if yaxis >= ai else i >= height - ai) and xaxislabel and axislabel:
|
||
strm += "y"
|
||
output = True
|
||
elif (
|
||
ylabellength
|
||
and (xaxislabel if xaxis < aj else j < xaxis - ylabellength and j + aj >= xaxis - ylabellength)
|
||
and (yaxis >= ai or i < height - ai)
|
||
and axistick
|
||
and axisunitslabel
|
||
):
|
||
strm += ylabelstrm
|
||
output = True
|
||
if ylabellength > aj:
|
||
j += ylabellength - aj
|
||
|
||
if not output:
|
||
dot = 0
|
||
color = 0
|
||
|
||
for k in range(min(aj, width - j)):
|
||
for l in range(min(ai, height - i)):
|
||
value = array[j + k][i + l]
|
||
if value:
|
||
if atype == type_types.braille:
|
||
dot += dotvalues[k][l]
|
||
elif atype in {
|
||
type_types.block,
|
||
type_types.block_quadrant,
|
||
type_types.separated_block_quadrant,
|
||
type_types.block_sextant,
|
||
type_types.separated_block_sextant,
|
||
type_types.block_octant,
|
||
}:
|
||
dot += 1 << (l * aj + k)
|
||
elif atype == type_types.histogram:
|
||
if not dot:
|
||
dot = (len(bars) - l) - 1
|
||
if color:
|
||
if value and color != value:
|
||
color = 1
|
||
else:
|
||
color = value
|
||
|
||
if color:
|
||
color -= 1
|
||
|
||
if color:
|
||
strm += outputcolor(color)
|
||
|
||
if atype == type_types.braille:
|
||
strm += dots[dot]
|
||
elif atype == type_types.block:
|
||
strm += blocks[dot]
|
||
elif atype == type_types.block_quadrant:
|
||
strm += blocks_quadrant[dot]
|
||
elif atype == type_types.separated_block_quadrant:
|
||
strm += separated_blocks_quadrant[dot]
|
||
elif atype == type_types.block_sextant:
|
||
strm += blocks_sextant[dot]
|
||
elif atype == type_types.separated_block_sextant:
|
||
strm += separated_blocks_sextant[dot]
|
||
elif atype == type_types.block_octant:
|
||
strm += blocks_octant[dot]
|
||
elif atype == type_types.histogram:
|
||
strm += bars[dot]
|
||
|
||
if color:
|
||
strm += outputcolor(color_types.default)
|
||
|
||
j += aj
|
||
|
||
if border:
|
||
strm += astyle[1]
|
||
|
||
if i < height - ai or border:
|
||
strm += "\n"
|
||
i += ai
|
||
|
||
if border:
|
||
strm += astyle[8] + (astyle[0] * awidth) + astyle[10]
|
||
|
||
print(strm, file=file)
|
||
|
||
return 0
|
||
|
||
|
||
def histogram(
|
||
height: int,
|
||
width: int,
|
||
xmin: float,
|
||
xmax: float,
|
||
ymin: float,
|
||
ymax: float,
|
||
aarray: Sequence[float],
|
||
border: bool = False,
|
||
axis: bool = True,
|
||
axislabel: bool = True,
|
||
axistick: bool = True,
|
||
axisunitslabel: bool = True,
|
||
xunits: units_types = units_types.fracts,
|
||
yunits: units_types = units_types.fracts,
|
||
style: style_types = style_types.light,
|
||
color: color_types = color_types.red,
|
||
title: Optional[str] = None,
|
||
file: TextIO = sys.stdout,
|
||
check: bool = True,
|
||
) -> int:
|
||
"""Convert one or more arrays to graph and output."""
|
||
if not aarray:
|
||
return 1
|
||
|
||
w = shutil.get_terminal_size()
|
||
|
||
if not height:
|
||
height = w.lines
|
||
|
||
if not width:
|
||
width = w.columns
|
||
|
||
if check:
|
||
if height > w.lines:
|
||
print(f"The height of the graph ({height}) is greater then the height of the terminal ({w.lines}).", file=sys.stderr)
|
||
return 1
|
||
|
||
if width > w.columns:
|
||
print(f"The width of the graph ({width}) is greater then the width of the terminal ({w.columns}).", file=sys.stderr)
|
||
return 1
|
||
|
||
ai, aj = densities[type_types.histogram]
|
||
|
||
height *= ai
|
||
width *= aj
|
||
|
||
if not xmin and not xmax:
|
||
xmin = min(aarray)
|
||
xmax = max(aarray)
|
||
|
||
if xmin >= xmax:
|
||
print("xmin must be less than xmax.", file=sys.stderr)
|
||
return 1
|
||
|
||
histogram = [0 for i in range(width)]
|
||
|
||
xstep = (xmax - xmin) / width
|
||
|
||
for x in aarray:
|
||
if xmin <= x < xmax:
|
||
index = int((x - xmin) / xstep)
|
||
histogram[index] += 1
|
||
|
||
if not ymin and not ymax:
|
||
ymin = min(histogram)
|
||
ymax = max(histogram)
|
||
|
||
if ymin >= ymax:
|
||
print("ymin must be less than ymax.", file=sys.stderr)
|
||
return 1
|
||
|
||
ystep = (ymax - ymin) / height
|
||
yaxis = ymax / ystep
|
||
|
||
aaarray = [[0 for j in range(height)] for i in range(width)]
|
||
|
||
acolor = color + 1
|
||
|
||
for x, ay in enumerate(histogram):
|
||
y = 0 if ay >= ymax else int(yaxis - (ay / ystep))
|
||
while y < yaxis and y < height:
|
||
aaarray[x][y] = acolor
|
||
y += 1
|
||
|
||
return graph(
|
||
height,
|
||
width,
|
||
xmin,
|
||
xmax,
|
||
ymin,
|
||
ymax,
|
||
aaarray,
|
||
border,
|
||
axis,
|
||
axislabel,
|
||
axistick,
|
||
axisunitslabel,
|
||
xunits,
|
||
yunits,
|
||
type_types.histogram,
|
||
style,
|
||
title,
|
||
file,
|
||
)
|
||
|
||
|
||
def plots(
|
||
height: int,
|
||
width: int,
|
||
xmin: float,
|
||
xmax: float,
|
||
ymin: float,
|
||
ymax: float,
|
||
aarrays: Sequence[Sequence[Sequence[float]]],
|
||
border: bool = False,
|
||
axis: bool = True,
|
||
axislabel: bool = True,
|
||
axistick: bool = True,
|
||
axisunitslabel: bool = True,
|
||
xunits: units_types = units_types.fracts,
|
||
yunits: units_types = units_types.fracts,
|
||
atype: type_types = type_types.braille,
|
||
mark: mark_types = mark_types.dot,
|
||
style: style_types = style_types.light,
|
||
color: color_types = color_types.red,
|
||
title: Optional[str] = None,
|
||
file: TextIO = sys.stdout,
|
||
check: bool = True,
|
||
) -> int:
|
||
"""Convert one or more arrays to graph and output."""
|
||
if not aarrays:
|
||
return 1
|
||
|
||
if not all(len(x) == 2 for aarray in aarrays for x in aarray):
|
||
print("Error: The arrays must have two columns.")
|
||
return 1
|
||
|
||
w = shutil.get_terminal_size()
|
||
|
||
if not height:
|
||
height = w.lines
|
||
|
||
if not width:
|
||
width = w.columns
|
||
|
||
if check:
|
||
if height > w.lines:
|
||
print(f"The height of the graph ({height}) is greater then the height of the terminal ({w.lines}).", file=sys.stderr)
|
||
return 1
|
||
|
||
if width > w.columns:
|
||
print(f"The width of the graph ({width}) is greater then the width of the terminal ({w.columns}).", file=sys.stderr)
|
||
return 1
|
||
|
||
ai, aj = densities[atype]
|
||
|
||
height *= ai
|
||
width *= aj
|
||
|
||
if not xmin and not xmax:
|
||
xmin = min(x for aarray in aarrays for x, y in aarray)
|
||
xmax = max(x for aarray in aarrays for x, y in aarray)
|
||
|
||
if not ymin and not ymax:
|
||
ymin = min(y for aarray in aarrays for x, y in aarray)
|
||
ymax = max(y for aarray in aarrays for x, y in aarray)
|
||
|
||
if xmin >= xmax:
|
||
print("xmin must be less than xmax.", file=sys.stderr)
|
||
return 1
|
||
|
||
if ymin >= ymax:
|
||
print("ymin must be less than ymax.", file=sys.stderr)
|
||
return 1
|
||
|
||
xstep = (xmax - xmin) / width
|
||
ystep = (ymax - ymin) / height
|
||
xaxis = width - xmax / xstep
|
||
yaxis = ymax / ystep
|
||
|
||
aaarray = [[0 for j in range(height)] for i in range(width)]
|
||
|
||
for j, aarray in enumerate(aarrays):
|
||
acolor = color + 1 if len(aarrays) == 1 else j % (len(colors) - 2) + 3
|
||
|
||
for x, y in aarray:
|
||
if xmin <= x < xmax and ymin <= y < ymax:
|
||
ax = int(x / xstep + xaxis)
|
||
ay = int(yaxis - y / ystep - 1)
|
||
|
||
for ix, iy in marks[mark]:
|
||
x = ax + ix
|
||
y = ay + iy
|
||
|
||
if x < width and y < height:
|
||
if aaarray[x][y]:
|
||
if aaarray[x][y] != acolor:
|
||
aaarray[x][y] = 1
|
||
else:
|
||
aaarray[x][y] = acolor
|
||
|
||
return graph(
|
||
height,
|
||
width,
|
||
xmin,
|
||
xmax,
|
||
ymin,
|
||
ymax,
|
||
aaarray,
|
||
border,
|
||
axis,
|
||
axislabel,
|
||
axistick,
|
||
axisunitslabel,
|
||
xunits,
|
||
yunits,
|
||
atype,
|
||
style,
|
||
title,
|
||
file,
|
||
)
|
||
|
||
|
||
def plot(
|
||
height: int,
|
||
width: int,
|
||
xmin: float,
|
||
xmax: float,
|
||
ymin: float,
|
||
ymax: float,
|
||
aarray: Sequence[Sequence[float]],
|
||
border: bool = False,
|
||
axis: bool = True,
|
||
axislabel: bool = True,
|
||
axistick: bool = True,
|
||
axisunitslabel: bool = True,
|
||
xunits: units_types = units_types.fracts,
|
||
yunits: units_types = units_types.fracts,
|
||
atype: type_types = type_types.braille,
|
||
mark: mark_types = mark_types.dot,
|
||
style: style_types = style_types.light,
|
||
color: color_types = color_types.red,
|
||
title: Optional[str] = None,
|
||
file: TextIO = sys.stdout,
|
||
check: bool = True,
|
||
) -> int:
|
||
"""Convert single array to graph and output."""
|
||
return plots(
|
||
height,
|
||
width,
|
||
xmin,
|
||
xmax,
|
||
ymin,
|
||
ymax,
|
||
(aarray,),
|
||
border,
|
||
axis,
|
||
axislabel,
|
||
axistick,
|
||
axisunitslabel,
|
||
xunits,
|
||
yunits,
|
||
atype,
|
||
mark,
|
||
style,
|
||
color,
|
||
title,
|
||
file,
|
||
check,
|
||
)
|
||
|
||
|
||
def functions(
|
||
height: int,
|
||
width: int,
|
||
xmin: float,
|
||
xmax: float,
|
||
ymin: float,
|
||
ymax: float,
|
||
afunctions: Sequence[Callable[[float], float]],
|
||
border: bool = False,
|
||
axis: bool = True,
|
||
axislabel: bool = True,
|
||
axistick: bool = True,
|
||
axisunitslabel: bool = True,
|
||
xunits: units_types = units_types.fracts,
|
||
yunits: units_types = units_types.fracts,
|
||
atype: type_types = type_types.braille,
|
||
style: style_types = style_types.light,
|
||
color: color_types = color_types.red,
|
||
title: Optional[str] = None,
|
||
file: TextIO = sys.stdout,
|
||
check: bool = True,
|
||
) -> int:
|
||
"""Convert one or more functions to graph and output."""
|
||
if not afunctions:
|
||
return 1
|
||
|
||
w = shutil.get_terminal_size()
|
||
|
||
if not height:
|
||
height = w.lines
|
||
|
||
if not width:
|
||
width = w.columns
|
||
|
||
if check:
|
||
if height > w.lines:
|
||
print(f"The height of the graph ({height}) is greater then the height of the terminal ({w.lines}).", file=sys.stderr)
|
||
return 1
|
||
|
||
if height > w.columns:
|
||
print(f"The width of the graph ({height}) is greater then the width of the terminal ({w.columns}).", file=sys.stderr)
|
||
return 1
|
||
|
||
ai, aj = densities[atype]
|
||
|
||
height *= ai
|
||
width *= aj
|
||
|
||
if xmin >= xmax:
|
||
print("xmin must be less than xmax.", file=sys.stderr)
|
||
return 1
|
||
|
||
if ymin >= ymax:
|
||
print("ymin must be less than ymax.", file=sys.stderr)
|
||
return 1
|
||
|
||
rows = width
|
||
|
||
xstep = (xmax - xmin) / width
|
||
ystep = (ymax - ymin) / height
|
||
xaxis = width - xmax / xstep
|
||
yaxis = ymax / ystep
|
||
xres = 2
|
||
|
||
array = [[0 for j in range(height)] for i in range(width)]
|
||
|
||
for j, function in enumerate(afunctions):
|
||
acolor = color + 1 if len(afunctions) == 1 else j % (len(colors) - 2) + 3
|
||
|
||
for i in (x / xres for x in range(rows * xres)):
|
||
x = i * xstep + xmin
|
||
y = function(x)
|
||
|
||
if xmin <= x < xmax and ymin <= y < ymax:
|
||
ax = int(x / xstep + xaxis)
|
||
ay = int(yaxis - y / ystep - 1)
|
||
|
||
if array[ax][ay]:
|
||
if array[ax][ay] != acolor:
|
||
array[ax][ay] = 1
|
||
else:
|
||
array[ax][ay] = acolor
|
||
|
||
return graph(
|
||
height,
|
||
width,
|
||
xmin,
|
||
xmax,
|
||
ymin,
|
||
ymax,
|
||
array,
|
||
border,
|
||
axis,
|
||
axislabel,
|
||
axistick,
|
||
axisunitslabel,
|
||
xunits,
|
||
yunits,
|
||
atype,
|
||
style,
|
||
title,
|
||
file,
|
||
)
|
||
|
||
|
||
def function(
|
||
height: int,
|
||
width: int,
|
||
xmin: float,
|
||
xmax: float,
|
||
ymin: float,
|
||
ymax: float,
|
||
afunction: Callable[[float], float],
|
||
border: bool = False,
|
||
axis: bool = True,
|
||
axislabel: bool = True,
|
||
axistick: bool = True,
|
||
axisunitslabel: bool = True,
|
||
xunits: units_types = units_types.fracts,
|
||
yunits: units_types = units_types.fracts,
|
||
atype: type_types = type_types.braille,
|
||
style: style_types = style_types.light,
|
||
color: color_types = color_types.red,
|
||
title: Optional[str] = None,
|
||
file: TextIO = sys.stdout,
|
||
check: bool = True,
|
||
) -> int:
|
||
"""Convert single function to function array and output."""
|
||
return functions(
|
||
height,
|
||
width,
|
||
xmin,
|
||
xmax,
|
||
ymin,
|
||
ymax,
|
||
(afunction,),
|
||
border,
|
||
axis,
|
||
axislabel,
|
||
axistick,
|
||
axisunitslabel,
|
||
xunits,
|
||
yunits,
|
||
atype,
|
||
style,
|
||
color,
|
||
title,
|
||
file,
|
||
check,
|
||
)
|