chromatic-python 0.1.0__py3-none-any.whl

chromatic/demo.py

@@ -0,0 +1,417 @@
+import functools
+import math
+import os
+import sys
+import time
+from os import PathLike
+from pathlib import PurePath
+from types import FunctionType
+from typing import Callable
+
+
+def escher_dragon_ascii():
+    """Displays the image-to-ASCII transform of 'Dragon' by M.C. Escher."""
+    from chromatic.ascii import ascii2img, img2ascii
+    from chromatic.data import UserFont, escher
+
+    input_img = escher()
+    font = UserFont.IBM_VGA_437_8X16
+    char_set = r"  ._-~+<vX♦'^Vx>|πΦ0Ω#$║╫"
+
+    ascii_str = img2ascii(
+        input_img,
+        font,
+        factor=240,
+        char_set=char_set,
+        sort_glyphs=True)
+
+    ascii_img = ascii2img(
+        ascii_str,
+        font,
+        font_size=16,
+        fg='white',
+        bg='black')
+
+    ascii_img.show()
+
+
+def escher_dragon_256color():
+    """Displays the image-to-ANSI transform of 'Dragon' by M.C. Escher in 8-bit color."""
+    from chromatic.ascii import ansi2img, img2ansi
+    from chromatic.data import UserFont, escher
+
+    input_img = escher()
+    font = UserFont.IBM_VGA_437_8X16
+
+    ansi_array = img2ansi(
+        input_img,
+        font,
+        factor=240,
+        ansi_type='8b',
+        equalize=True)
+
+    ansi_img = ansi2img(
+        ansi_array,
+        font,
+        font_size=16)
+
+    ansi_img.show()
+
+
+def butterfly_16color():
+    """Displays image-to-ANSI transform of 'Spider Lily & Papilio xuthus' in 4-bit color.
+
+    Good ol' C-x M-c M-butterfly...
+    """
+    from chromatic.color import ansicolor4Bit
+    from chromatic.ascii import ansi2img, img2ansi
+    from chromatic.data import UserFont, butterfly
+
+    input_img = butterfly()
+
+    font = UserFont.IBM_VGA_437_8X16
+
+    char_set = r"'·,•-_→+<>ⁿ*%⌂7√Iï∞πbz£9yîU{}1αHSw♥æ?GX╕╒éà⌡MF╝╩ΘûǃQ½☻Ŷ┤▄╪║▒█"
+
+    ansi_array = img2ansi(
+        input_img,
+        font,
+        factor=200,
+        char_set=char_set,
+        ansi_type=ansicolor4Bit)
+
+    ansi_img = ansi2img(
+        ansi_array,
+        font,
+        font_size=16)
+
+    ansi_img.show()
+
+
+def butterfly_truecolor():
+    """Displays the image-to-ANSI transform of 'Spider Lily & Papilio xuthus' in 24-bit color."""
+    from chromatic.ascii import ansi2img, img2ansi
+    from chromatic.data import UserFont, butterfly
+
+    input_img = butterfly()
+
+    font = UserFont.IBM_VGA_437_8X16
+
+    ansi_array = img2ansi(
+        input_img,
+        font,
+        factor=200,
+        ansi_type='24b',
+        equalize='white_point')
+
+    ansi_img = ansi2img(
+        ansi_array,
+        font,
+        font_size=16)
+
+    ansi_img.show()
+
+
+def butterfly_randcolor():
+    from chromatic.ascii import ansi2img, img2ansi
+    from chromatic.color import randcolor, rgb2hsv, hsv2rgb, Color
+    from chromatic.data import UserFont, butterfly
+
+    input_img = butterfly()
+
+    font = UserFont.IBM_VGA_437_8X16
+
+    ansi_array = img2ansi(
+        input_img,
+        font,
+        factor=200,
+        ansi_type='8b',
+        equalize='white_point')
+
+    for row in range(len(ansi_array)):
+        for idx, cs in enumerate(ansi_array[row]):
+            if (fg := cs.fg) is not None:
+                _, _, v = rgb2hsv(fg.rgb)
+                h, s, _ = rgb2hsv(randcolor().rgb)
+                ansi_array[row][idx] = cs.recolor(fg=Color.from_rgb(hsv2rgb((h, s, v))))
+
+    ansi_img = ansi2img(
+        ansi_array,
+        font,
+        font_size=16)
+
+    ansi_img.show()
+
+
+def goblin_virus_truecolor():
+    """`G-O-B-L-I-N VIRUS <https://imgur.com/n0Mng2P>`__"""
+    from chromatic.ascii import ansi2img, img2ansi
+    from chromatic.data import UserFont, goblin_virus
+
+    input_img = goblin_virus()
+
+    font = UserFont.IBM_VGA_437_8X16
+
+    char_set = r'  .-|_⌐¬^:()═+<>v≥≤«*»x└┘π╛╘┴┐┌┬╧╚╙X╒╜╨#0╓╝╩╤╥│╔┤├╞╗╦┼╪║╟╠╫╣╬░▒▓█▄▌▐▀'
+
+    ansi_array = img2ansi(
+        input_img,
+        font,
+        factor=200,
+        char_set=char_set,
+        ansi_type='24b',
+        equalize=False)
+
+    ansi_img = ansi2img(
+        ansi_array,
+        font,
+        font_size=16)
+
+    ansi_img.show()
+
+
+def named_colors():
+    from chromatic.color.palette import display_named_colors, ColorNamespace
+    from chromatic.color.colorconv import rgb2hsv, rgb2lab
+
+    print(f"{'.'.join([ColorNamespace.__module__, ColorNamespace.__name__])}:")
+    named = display_named_colors()
+    whites = [0]
+    for idx, n in enumerate(named):
+        hsv = rgb2hsv(n.fg.rgb)
+        if all(
+                map(
+                    lambda i, x: math.isclose(hsv[i], x, abs_tol=.16),
+                    (-1, 1), (1, 0))):
+            if idx - whites[-1] < 4:
+                whites.pop()
+            whites.append(idx)
+    whites.append(-1)
+    buffer = []
+    for start, stop in zip(whites, whites[1:]):
+        xs = sorted(
+            named[start + 1 if start else None:stop + 1 if ~stop else None],
+            key=lambda x: rgb2lab(x.fg.rgb))
+        buffer.append(xs)
+    for ln in buffer:
+        print(' | '.join(ln))
+
+
+def color_table():
+    """Print foreground / background combinations in each ANSI format.
+
+    A handful of stylistic SGR parameters are displayed as well.
+    """
+    from chromatic.color import (
+        ColorStr,
+        SgrParameter,
+        ansicolor24Bit,
+        ansicolor4Bit,
+        ansicolor8Bit
+    )
+    from chromatic.color.palette import ColorNamespace
+
+    color_ns = ColorNamespace()
+    ansi_types = [ansicolor4Bit, ansicolor8Bit, ansicolor24Bit]
+    colors = [
+        color_ns.BLACK,
+        color_ns.WHITE,
+        color_ns.RED,
+        color_ns.ORANGE,
+        color_ns.YELLOW,
+        color_ns.GREEN,
+        color_ns.BLUE,
+        color_ns.INDIGO,
+        color_ns.PURPLE]
+    colors_dict = {v.name.title(): v for v in colors}
+    spacing = max(map(len, colors_dict)) + 1
+    fg_colors = [ColorStr(
+        f"{c.name.title(): ^{spacing}}",
+        color_spec=dict(fg=c),
+        ansi_type=ansicolor24Bit)
+        for c in colors]
+    bg_colors = [
+                    ColorStr().recolor(bg=None)
+                ] + [c.recolor(fg=None, bg=c.fg) for c in fg_colors]
+    pad = spacing - 1
+    print(
+        '|'.join(
+            [f"{'4bit': ^{pad}}",
+             f"{'8bit': ^{pad}}",
+             f"{'24bit': >{pad}}"]))
+    for row in fg_colors:
+        for col in bg_colors:
+            for typ in ansi_types:
+                print(row.as_ansi_type(typ).recolor(bg=col.bg), end='\x1b[0m')
+        print()
+    print('\nstyles:')
+    print()
+    style_params = [
+        SgrParameter.BOLD,
+        SgrParameter.ITALICS,
+        SgrParameter.CROSSED_OUT,
+        SgrParameter.ENCIRCLED,
+        SgrParameter.SINGLE_UNDERLINE,
+        SgrParameter.DOUBLE_UNDERLINE,
+        SgrParameter.NEGATIVE]
+    for style in style_params:
+        print(
+            ColorStr('.'.join([SgrParameter.__qualname__, style.name]))
+            .update_sgr(style),
+            end='\x1b[0m' + (' ' * 4))
+    print()
+
+
+def glyph_comparisons(__output_dir: str | PathLike[str] = None):
+    from skimage.metrics import mean_squared_error
+    from numpy import ndarray
+    from chromatic.ascii import cp437_printable
+    from chromatic import get_glyph_masks
+    from chromatic.data import UserFont
+    from random import choices as get_random
+
+    def _find_best_matches(glyph_masks1: dict[str, ndarray],
+                           glyph_masks2: dict[str, ndarray]) -> dict[str, str]:
+        best_matches = {}
+        for char1, mask1 in glyph_masks1.items():
+            best_char = None
+            best_score = float('inf')
+            for char2, mask2 in glyph_masks2.items():
+                score = mean_squared_error(mask1, mask2)
+                if score < best_score:
+                    best_score = score
+                    best_char = char2
+            best_matches[char1] = best_char
+        return best_matches
+
+    if __output_dir and not os.path.isdir(__output_dir):
+        raise NotADirectoryError(
+            __output_dir)
+    user_fonts = [pair := (UserFont.IBM_VGA_437_8X16, UserFont.CONSOLAS), pair[::-1]]
+    trans_table = str.maketrans({']': None, '0': ' ', '[': ' '})
+    char_set = cp437_printable()
+    separator = '#' * 100
+    for font1, font2 in user_fonts:
+        glyph_masks_1 = get_glyph_masks(font1, char_set, dist_transform=True)
+        glyph_masks_2 = get_glyph_masks(font2, char_set, dist_transform=True)
+        best_matches_ = _find_best_matches(glyph_masks_1, glyph_masks_2)
+        txt = ''.join(
+            '->'.center(32, ' ').join(['{}'] * 2).format(
+                f"{font1.name}"
+                f"[{input_char!r}, {input_char.encode('unicode_escape').decode()!r}]",
+                f"{font2.name}"
+                f"[{matched_char!r}, {matched_char.encode('unicode_escape').decode()!r}]")
+            .center(100, ' ')
+            + '\n\n'
+            + '\n'.join(
+                ''.join
+                (z).translate(trans_table)
+                for z in zip(
+                    f'{glyph_masks_1[input_char].astype(int)}\n'.splitlines(),
+                    f'{glyph_masks_2[matched_char].astype(int)}\n'.splitlines()[1:]))
+            + separator.join(['\n'] * 2) for input_char, matched_char in best_matches_.items())
+        if __output_dir is not None:
+            fname = (PurePath(__output_dir) /
+                     f"{'_to_'.join(font.name.lower() for font in (font1, font2))}.txt")
+            with open(fname, 'w', encoding='utf-8') as f:
+                f.write(txt)
+        else:
+            for glyph in get_random(txt.split(separator), k=len(char_set) // 2):
+                print(separator + glyph)
+
+
+class _time_wrapper[** P, R]:
+
+    def __init__(
+        self,
+        func: Callable[P, R] | FunctionType | type = None
+    ):
+        self.func = func
+        if self.func is not None:
+            functools.update_wrapper(self, self.func)
+
+    def __call__(self, *args: P.args, **kwargs: P.kwargs) -> R:
+        if self.func is not None:
+            start = time.perf_counter()
+            result = self.func(*args, **kwargs)
+            stop = time.perf_counter()
+            print(f"Total execution time: {self._delta(start, stop)}")
+            return result
+        else:
+            self.func = args[0]
+            functools.update_wrapper(self, self.func)
+            return self
+
+    @staticmethod
+    def _delta(start: float, stop: float) -> str:
+        delta = stop - start
+        mag, fmt = min(
+            [(1, 's'), (1e-3, 'ms'), (1e-6, 'μs'), (1e-9, 'ns'), (1e-12, 'ps')],
+            key=lambda x: abs(math.log10(x[0]) - math.log10(delta)))
+        delta *= 1 / mag
+        return f"{delta:.3f} {fmt}"
+
+
+def main():
+    demo_globals = dict(globals())
+    demo_globals.pop('main')
+    from inspect import getargs
+
+    global_func_enum = dict(
+        enumerate(
+            sorted(
+                k for k, v in demo_globals.items() if
+                isinstance(v, FunctionType) and not v.__name__.startswith('_'))))
+    safe_funcs = {-1: exit}
+    choices = [f'[{x[0]}]: {x[1].name}' for x in safe_funcs.items()]
+    names = []
+    for k, v in global_func_enum.items():
+        if not any(getargs(demo_globals[v].__code__)):
+            if safe_funcs.get(k - 1) is None:
+                k_val = list(safe_funcs).pop() + 1
+            else:
+                k_val = k
+            safe_funcs[k_val] = globals()[v]
+            choices.append(f"[{k_val}]: {v}")
+            names.append(v)
+
+    def _check_user_input(user_key: str):
+        if user_key.strip('-').isdigit():
+            if (k := int(user_key)) in safe_funcs:
+                return k
+        if (s := user_key.strip().replace(' ', '_').casefold()) in names:
+            return next(i for i, v in enumerate(names) if v == s)
+        return
+
+    selection = None
+    if len(sys.argv) > 1:
+        key = sys.argv[1]
+        if key.casefold() == '-h'.casefold():
+            xs = ['Run one of the following demo functions:',
+                  *('{}:\n\t{}'.format(
+                      n, '\n'.join(
+                          filter(None, (globals()[n].__doc__ or '...').splitlines())).strip())
+                      for n in names)]
+            print(f'\n{'\n\n'.join(xs)}')
+            exit()
+        selection = _check_user_input(key)
+    if selection is None:
+        print('\n'.join(choices))
+    while selection not in safe_funcs:
+        try:
+            selection = _check_user_input(input(f"Select a demo function:\t"))
+        except ValueError:
+            pass
+        except KeyboardInterrupt:
+            exit()
+    try:
+        if selection == -1:
+            exit()
+        print(f"Running {names[selection]!r}...\n")
+    except KeyError:
+        pass
+    _time_wrapper(safe_funcs[selection])()
+
+
+if __name__ == '__main__':
+    main()

chromatic_python-0.1.0.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 crypt0lith
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

chromatic_python-0.1.0.dist-info/METADATA

@@ -0,0 +1,39 @@
+Metadata-Version: 2.1
+Name: chromatic-python
+Version: 0.1.0
+Summary: ANSI art image processing and colored terminal text
+Author: crypt0lith
+License: MIT License
+        
+        Copyright (c) 2024 crypt0lith
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+Keywords: ansi,ascii,art,font,image,terminal,parser
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: fonttools~=4.51.0
+Requires-Dist: networkx~=3.4.2
+Requires-Dist: numpy~=2.1.1
+Requires-Dist: opencv-python~=4.10.0.84
+Requires-Dist: pillow~=10.4.0
+Requires-Dist: scikit-image~=0.25.0rc1
+Requires-Dist: scikit-learn~=1.5.2
+Requires-Dist: scipy~=1.14.1

chromatic_python-0.1.0.dist-info/RECORD

@@ -0,0 +1,26 @@
+chromatic/__init__.py,sha256=0wwCYq16TA-FUJTMt9KxcaA09owmh5yrtfNXCB1fg3s,589
+chromatic/_typing.py,sha256=6AYJx2dsRhLOtOwy3TTLSYLcBdCGaqKwR46Z41u9bbY,6085
+chromatic/demo.py,sha256=Vi-TFpw1ARnZRbfaCca1bT4-FbtrjXejTOspycr8eU8,13151
+chromatic/ascii/__init__.py,sha256=Ic3bqfqr0uAJd6vB4kToDookvZVr8Nrkyy3speiH5oM,164
+chromatic/ascii/_array.py,sha256=1vR_LsGvQrAUmwIBBeFfmyEKciI7chdV_0kr2Q-gAdw,36197
+chromatic/ascii/_curses.py,sha256=9zUeBrylcMXD-R0n5KBVqafAecd6lrQX1Yxjp-BrN98,3719
+chromatic/ascii/_glyph_proc.py,sha256=rqtx6zLxwqqHa1YzuBhL-zhr3LptaHZBnL6ow9W3eI0,2840
+chromatic/color/__init__.py,sha256=8dnpWqbmEciFgJrQpUC0gV9M5D1bKIF7d9Sl4mj0x7g,185
+chromatic/color/colorconv.py,sha256=Ie6P_MUIXk6KcxyIJMw7HuGZWg9sBZVzgGYX4XRMDfA,8581
+chromatic/color/core.py,sha256=K3JpWxsenRqd2IoNRnJRSxK6cpSpGV_M8wVqaQxJfmI,58488
+chromatic/color/core.pyi,sha256=0bREo8eV5LNiA-a014w0C1yML7F3MG3_Vj_XcC6bYAs,11310
+chromatic/color/palette.py,sha256=RTEj-YO20ZlVP2D9lhlge2v-ef5ENUyxHubAo0Ji5Og,23450
+chromatic/color/palette.pyi,sha256=cj9Lmgw6MVCDY8Ew7wc9CvyWP6G4ZqRZnXJgOizBAO8,8780
+chromatic/data/__init__.py,sha256=TCSacSbkvVExjufkNxarEYCSpKYnybm-gljfA_eumG4,6581
+chromatic/data/__init__.pyi,sha256=AZGa46n1Hck3xHN91Mmm7Qibg--CYweKJMMFPHH6yY8,556
+chromatic/data/fonts/IBM_VGA_437_8x16.ttf,sha256=qMdn-pJWJNKNmHnDoDqGIE94vOTezaCiBv0VK92QbJQ,50124
+chromatic/data/fonts/consolas.ttf,sha256=xubOgRn91H7GpUSaCOLSrX9B6gMUOq4ZMGjtn6WOrrw,459180
+chromatic/data/images/butterfly.jpg,sha256=rnSfMVotSgpTNRPQ9CzYpDp8S8lkCAJ5ZySRrvyDe7I,448399
+chromatic/data/images/escher.png,sha256=FvAC-WOkUVIqHBxasfgCJbH1CN1IwLE-JyMiATOEKFc,132667
+chromatic/data/images/goblin_virus.png,sha256=ygs19t4ZeZYsbiRFXjRyEz8OtgC9Zyl8owX0O-5n40o,8849
+chromatic/data/images/hotdog.jpg,sha256=b0vnoQCarhq-gFu3df512qPTvauvF1Oa0oXVO9msEFc,60823
+chromatic_python-0.1.0.dist-info/LICENSE,sha256=e7GjzmO7L12JDai3XRWHD8PgZvFmzNceEztB6d-9kuA,1086
+chromatic_python-0.1.0.dist-info/METADATA,sha256=VhHfYHh_7GjWE67GYyCQOOuny1tvlgyup780MAC72e8,1869
+chromatic_python-0.1.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+chromatic_python-0.1.0.dist-info/top_level.txt,sha256=wjzxcxfjO8I4u22BS8wL67Bq64c59kbZCqQ--Fc_Mqw,10
+chromatic_python-0.1.0.dist-info/RECORD,,

chromatic_python-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.6.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

chromatic_python-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+chromatic