mirror of
https://github.com/1j01/textual-paint.git
synced 2024-12-22 14:21:33 +03:00
77 lines
2.4 KiB
Python
Executable File
77 lines
2.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import math
|
|
from typing import TextIO
|
|
|
|
# ANSI escape codes for truecolor
|
|
CSI = '\u001b['
|
|
RESET = CSI + '0m'
|
|
|
|
# Glyphs to use for the gradient
|
|
GLYPHS = ['🌸', '🌷', '🌹', '🌺', '🌻', '🌼']
|
|
|
|
def spiral(m: tuple[float, float]) -> float:
|
|
r = math.sqrt(m[0]**2 + m[1]**2) * .05
|
|
a = math.atan2(m[1], m[0])
|
|
v = math.sin(100. * (math.sqrt(r) - 0.02 * a))
|
|
return max(0., min(1., v))
|
|
|
|
def generate_ansi_art(width: int, height: int, file: TextIO) -> None:
|
|
# Calculate the center coordinates of the image
|
|
center_x = width // 2
|
|
center_y = height // 2
|
|
|
|
for y in range(height):
|
|
for x in range(width):
|
|
# Calculate the coordinates relative to the center
|
|
rel_x = x - center_x
|
|
rel_y = y - center_y
|
|
|
|
# Evaluate the spiral function
|
|
scale = 0.1
|
|
v = spiral((rel_x * scale, rel_y * scale))
|
|
|
|
# Calculate the color based on the spiral value
|
|
r = int(v * 255)
|
|
g = int(((1 - v) * 255)*0.8)
|
|
b = int(v * 255)
|
|
|
|
# Add linear gradient to the color
|
|
r = int(r * (1 - y / height))
|
|
# g = int(g * (1 - y / height))
|
|
# b = int(b * (1 - y / height))
|
|
|
|
# Generate the truecolor escape code
|
|
# background:
|
|
color = CSI + f'48;2;{r};{g};{b}m'
|
|
# and foreground:
|
|
# THIS WON'T DO MUCH for emoji, but it's here to INCREASE FILE SIZE
|
|
color += CSI + f'38;2;{r//10};{g//10};{b//10}m'
|
|
|
|
# Calculate the index of the glyph based on visual weight
|
|
glyph_index = int(v * (len(GLYPHS) - 1))
|
|
|
|
# Write the colored glyph to the file
|
|
file.write(color + GLYPHS[glyph_index])
|
|
|
|
# Reset the color at the end of each row and add a newline character
|
|
file.write(RESET + '\n')
|
|
|
|
# Set the size of the ANSI art
|
|
width = 80
|
|
height = 24
|
|
|
|
# Generate and write the ANSI art to a file
|
|
file_path = os.path.join(os.path.dirname(__file__), '../samples/gradient_test.ans')
|
|
file_path = os.path.abspath(file_path)
|
|
with open(file_path, 'w', encoding='utf-8') as file:
|
|
generate_ansi_art(width, height, file)
|
|
|
|
# Print the art to the terminal
|
|
with open(file_path, 'r', encoding='utf-8') as file:
|
|
print(file.read())
|
|
|
|
# Print the path to the file, and resulting file size
|
|
print(f'Wrote ANSI art to {file_path} ({os.path.getsize(file_path)} bytes)')
|