2025-07-01

2026-03-17 14:30:01 -06:00
parent f9a22056dd
commit 62b5978595
4579 changed files with 1257472 additions and 0 deletions
@@ -0,0 +1,119 @@
+'''
+Copyright (C) 2023 CG Cookie
+http://cgcookie.com
+hello@cgcookie.com
+
+Created by Jonathan Denning
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+'''
+
+from __future__ import annotations
+from typing import Any, Iterable
+from contextlib import contextmanager
+from ..ext import termcolor
+import random
+import time
+
+def colored(
+    text: str,
+    color: str | None = None,
+    *,
+    highlight: str | None = None,
+    attributes: Iterable[str] | None = None,
+    no_color: bool | None = None,
+    force_color: bool | None = None,
+) -> str:
+    return termcolor.colored(
+        text,
+        color=color,
+        on_color=f'on_{highlight}' if highlight else None,
+        attrs=attributes,
+        no_color=no_color,
+        force_color=force_color,
+    )
+
+def cprint(
+    text: str,
+    *,
+    color: str | None = None,
+    highlight: str | None = None,
+    attributes: Iterable[str] | None = None,
+    no_color: bool | None = None,
+    force_color: bool | None = None,
+    **kwargs: Any,
+) -> None:
+    print(
+        colored(
+            text,
+            color=color,
+            highlight=highlight,
+            attributes=attributes,
+            no_color=no_color,
+            force_color=force_color,
+        ),
+        **kwargs,
+    )
+
+def boxed(*lines, title=None, prefix='', margin='', pad=' ', sides='single', color=None, highlight=None, attributes=None, wrap=120, indent=4):
+    lines = [line for lines_ in lines for line in lines_.splitlines()]
+    # https://www.w3.org/TR/xml-entity-names/025.html
+    tl,tm,tr,lm,rm,bl,bm,br,lt,rt = {
+        'single': '┌─┐││└─┘┤├',
+        'double': '╔═╗║║╚═╝╡╞',
+    }[sides]
+    if title:
+        title = f'{tm}{lt} {title} {rt}{tm}'
+        title_width = len(title)
+    else:
+        title_width = 0
+    pad_width = len(pad) * 2
+    width = max(max(len(line) for line in lines), title_width)
+    if wrap and width > wrap:
+        width = wrap
+        wrapped_lines = []
+        for line in lines:
+            cur_indent = len(line) - len(line.lstrip()) + indent
+            first = True
+            while True:
+                if first: first = False
+                else:     line = (' '*cur_indent) + line
+                wrapped_lines.append(line[:wrap])
+                line = line[wrap:]
+                if not line: break
+        lines = wrapped_lines
+    if prefix: print(prefix, end='')
+    if title:
+        cprint(f'{margin}{tl}{title}{tm*(width+pad_width-len(title))}{tr}{margin}', color=color, highlight=highlight, attributes=attributes)
+    else:
+        cprint(f'{margin}{tl}{tm*(width+pad_width)}{tr}{margin}', color=color, highlight=highlight, attributes=attributes)
+    for line in lines:
+        if prefix: print(prefix, end='')
+        cprint(f'{margin}{lm}{pad}{line}{" "*(width - len(line))}{pad}{rm}{margin}', color=color, highlight=highlight, attributes=attributes)
+    if prefix: print(prefix, end='')
+    cprint(f'{margin}{bl}{bm*(width+pad_width)}{br}{margin}', color=color, highlight=highlight, attributes=attributes)
+
+sprint_data = {
+    'width':  5,
+    'index': -1,
+    'time':  None,
+}
+def sprint(*args):
+    global sprint_data
+    sprint_data['index'] = (sprint_data['index']+1) % sprint_data['width']
+    d = time.time() - sprint_data['time'] if sprint_data['time'] else 0
+    sprint_data['time'] = time.time()
+    m = ' '*sprint_data['index'] + '*' + ' '*(sprint_data['width']-sprint_data['index']-1)
+    s = " ".join(f'{arg}' for arg in args)
+    print(f'[{m}] {d:0.4f}s | {s}')