pyblock/pybitblock/terminal_matrix/matrix.py
2021-12-03 15:59:28 -06:00

123 lines
3 KiB
Python

#! /usr/bin/env python3
# Author: Joao S. O. Bueno
# gwidion@gmail.com
# GPL. v3.0
MAX_CASCADES = 600
MAX_COLS = 20
FRAME_DELAY = 0.03
MAX_SPEED = 5
import shutil, sys, time
from random import choice, randrange, paretovariate
CSI = "\x1b["
pr = lambda command: print("\x1b[", command, sep="", end="")
getchars = lambda start, end: [chr(i) for i in range(start, end)]
black, green, white = "30", "32", "37"
latin = getchars(0x30, 0x80)
greek = getchars(0x390, 0x3d0)
hebrew = getchars(0x5d0, 0x5eb)
cyrillic = getchars(0x400, 0x50)
chars= latin + greek + hebrew + cyrillic
def pareto(limit):
scale = lines // 2
number = (paretovariate(1.16) - 1) * scale
return max(0, limit - number)
def init():
global cols, lines
cols, lines = shutil.get_terminal_size()
pr("?25l") # Hides cursor
pr("s") # Saves cursor position
def end():
pr("m") # reset attributes
pr("2J") # clear screen
pr("u") # Restores cursor position
pr("?25h") # Show cursor
def print_at(char, x, y, color="", bright="0"):
pr("%d;%df" % (y, x))
pr(bright + ";" + color + "m")
print(char, end="", flush=True)
def update_line(speed, counter, line):
counter += 1
if counter >= speed:
line += 1
counter = 0
return counter, line
def cascade(col):
speed = randrange(1, MAX_SPEED)
espeed = randrange(1, MAX_SPEED)
line = counter = ecounter = 0
oldline = eline = -1
erasing = False
bright = "1"
limit = pareto(lines)
while True:
counter, line = update_line(speed , counter, line)
if randrange(10 * speed) < 1:
bright = "0"
if line > 1 and line <= limit and oldline != line:
print_at(choice(chars),col, line-1, green, bright)
if line < limit:
print_at(choice(chars),col, line, white, "1")
if erasing:
ecounter, eline = update_line(espeed, ecounter, eline)
print_at(" ",col, eline, black)
else:
erasing = randrange(line + 1) > (lines / 2)
eline = 0
yield None
oldline = line
if eline >= limit:
print_at(" ", col, oldline, black)
break
def main():
cascading = set()
added_new = True
while True:
while add_new(cascading): pass
stopped = iterate(cascading)
sys.stdout.flush()
cascading.difference_update(stopped)
time.sleep(FRAME_DELAY)
def add_new(cascading):
if randrange(MAX_CASCADES + 1) > len(cascading):
col = randrange(cols)
for i in range(randrange(MAX_COLS)):
cascading.add(cascade((col + i) % cols))
return True
return False
def iterate(cascading):
stopped = set()
for c in cascading:
try:
next(c)
except StopIteration:
stopped.add(c)
return stopped
def doit():
try:
init()
main()
except KeyboardInterrupt:
pass
finally:
end()
if __name__=="__main__":
doit()