Initial commit
This commit is contained in:
commit
0087e815f8
GPG Key ID:
6758ACE46AA2A849
|
|
@ -0,0 +1 @@
|
|||
__pycache__
|
||||
|
|
@ -0,0 +1,418 @@
|
|||
#! /usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
# vim:fenc=utf-8
|
||||
#
|
||||
# Copyright © 2019 pavle <pavle.portic@tilda.center>
|
||||
#
|
||||
# Distributed under terms of the BSD-3-Clause license.
|
||||
|
||||
|
||||
import numpy as np
|
||||
import curses
|
||||
|
||||
MAX_ITERS = 100
|
||||
COLOR_COUNT = 101
|
||||
ZOOM_LEVEL = 0
|
||||
X_SCALE = 1.75
|
||||
Y_SCALE = 1
|
||||
X_MID = -0.75
|
||||
Y_MID = 0
|
||||
|
||||
|
||||
def get_points(cols, rows):
|
||||
x_min = X_MID - X_SCALE * (1 / (2 ** ZOOM_LEVEL))
|
||||
x_max = X_MID + X_SCALE * (1 / (2 ** ZOOM_LEVEL))
|
||||
y_min = Y_MID - Y_SCALE * (1 / (2 ** ZOOM_LEVEL))
|
||||
y_max = Y_MID + Y_SCALE * (1 / (2 ** ZOOM_LEVEL))
|
||||
x = np.linspace(x_min, x_max, cols)
|
||||
y = np.linspace(y_min, y_max, rows)
|
||||
return x, y
|
||||
|
||||
|
||||
def calculate_point(x, y):
|
||||
z = 0
|
||||
for i in range(MAX_ITERS):
|
||||
z = z ** 2 + complex(x, y)
|
||||
if abs(z) >= 2:
|
||||
return i
|
||||
|
||||
return -1
|
||||
|
||||
|
||||
def calculate_set(cols, rows):
|
||||
x, y = get_points(cols, rows)
|
||||
matrix = np.zeros([rows, cols])
|
||||
for i in range(rows):
|
||||
for j in range(cols):
|
||||
matrix[i, j] = calculate_point(x[j], y[i])
|
||||
|
||||
return matrix
|
||||
|
||||
|
||||
def print_set(screen, cols, rows, matrix, char):
|
||||
screen.clear()
|
||||
for i in range(rows):
|
||||
for j in range(cols):
|
||||
iteration = matrix[i, j]
|
||||
if iteration == -1:
|
||||
screen.addstr(i, j, ' ', curses.color_pair(1))
|
||||
else:
|
||||
color_index = (int(iteration) + 2) % COLOR_COUNT
|
||||
screen.addstr(i, j, ' ', curses.color_pair(color_index))
|
||||
|
||||
screen.addstr(rows, 0, f'Iterations: {MAX_ITERS}, Zoom: {ZOOM_LEVEL}, Char: {char}', curses.color_pair(COLOR_COUNT))
|
||||
screen.refresh()
|
||||
|
||||
|
||||
def handle_keyboard(char):
|
||||
global X_MID
|
||||
global Y_MID
|
||||
global ZOOM_LEVEL
|
||||
global MAX_ITERS
|
||||
|
||||
if char == 113:
|
||||
return True
|
||||
elif char == 258: # Down
|
||||
Y_MID += Y_SCALE * (1 / (2 ** (ZOOM_LEVEL + 2)))
|
||||
return False
|
||||
elif char == 259: # Up
|
||||
Y_MID -= Y_SCALE * (1 / (2 ** (ZOOM_LEVEL + 2)))
|
||||
return False
|
||||
elif char == 260: # Left
|
||||
X_MID -= X_SCALE * (1 / (2 ** (ZOOM_LEVEL + 2)))
|
||||
return False
|
||||
elif char == 261: # Right
|
||||
X_MID += X_SCALE * (1 / (2 ** (ZOOM_LEVEL + 2)))
|
||||
return False
|
||||
elif char == 40: # Zoom out
|
||||
if ZOOM_LEVEL > 0:
|
||||
ZOOM_LEVEL -= 1
|
||||
return False
|
||||
elif char == 41: # Zoom in
|
||||
ZOOM_LEVEL += 1
|
||||
return False
|
||||
elif char == 43: # Iterations up
|
||||
MAX_ITERS += 1
|
||||
return False
|
||||
elif char == 45: # Iterations down
|
||||
MAX_ITERS -= 1
|
||||
return False
|
||||
|
||||
|
||||
def init_colors():
|
||||
global COLOR_COUNT
|
||||
COLOR_COUNT = 21
|
||||
curses.init_color(1, 0, 0, 0)
|
||||
curses.init_color(2, 1000, 0, 0)
|
||||
curses.init_color(3, 1000, 333, 0)
|
||||
curses.init_color(4, 1000, 666, 0)
|
||||
curses.init_color(5, 1000, 1000, 0)
|
||||
curses.init_color(6, 666, 1000, 0)
|
||||
curses.init_color(7, 333, 1000, 0)
|
||||
curses.init_color(8, 0, 1000, 0)
|
||||
curses.init_color(9, 0, 1000, 333)
|
||||
curses.init_color(10, 0, 1000, 666)
|
||||
curses.init_color(11, 0, 1000, 1000)
|
||||
curses.init_color(12, 0, 666, 1000)
|
||||
curses.init_color(13, 0, 333, 1000)
|
||||
curses.init_color(14, 0, 0, 1000)
|
||||
curses.init_color(15, 333, 0, 1000)
|
||||
curses.init_color(16, 666, 0, 1000)
|
||||
curses.init_color(17, 1000, 0, 1000)
|
||||
curses.init_color(18, 1000, 0, 666)
|
||||
curses.init_color(19, 1000, 0, 333)
|
||||
curses.init_color(20, 1000, 0, 0)
|
||||
curses.init_color(21, 50, 50, 50)
|
||||
curses.init_color(22, 800, 800, 800)
|
||||
curses.init_pair(1, 0, 1)
|
||||
curses.init_pair(2, 0, 2)
|
||||
curses.init_pair(3, 0, 3)
|
||||
curses.init_pair(4, 0, 4)
|
||||
curses.init_pair(5, 0, 5)
|
||||
curses.init_pair(6, 0, 6)
|
||||
curses.init_pair(7, 0, 7)
|
||||
curses.init_pair(8, 0, 8)
|
||||
curses.init_pair(9, 0, 9)
|
||||
curses.init_pair(10, 0, 10)
|
||||
curses.init_pair(11, 0, 11)
|
||||
curses.init_pair(12, 0, 12)
|
||||
curses.init_pair(13, 0, 13)
|
||||
curses.init_pair(14, 0, 14)
|
||||
curses.init_pair(15, 0, 15)
|
||||
curses.init_pair(16, 0, 16)
|
||||
curses.init_pair(17, 0, 17)
|
||||
curses.init_pair(18, 0, 18)
|
||||
curses.init_pair(19, 0, 19)
|
||||
curses.init_pair(20, 0, 20)
|
||||
curses.init_pair(21, 22, 21)
|
||||
|
||||
|
||||
def init_monochrome():
|
||||
global COLOR_COUNT
|
||||
COLOR_COUNT = 21
|
||||
curses.init_color(1, 0, 0, 0)
|
||||
curses.init_color(2, 50, 50, 50)
|
||||
curses.init_color(3, 100, 100, 100)
|
||||
curses.init_color(4, 150, 150, 150)
|
||||
curses.init_color(5, 200, 200, 200)
|
||||
curses.init_color(6, 250, 250, 250)
|
||||
curses.init_color(7, 300, 300, 300)
|
||||
curses.init_color(8, 350, 350, 350)
|
||||
curses.init_color(9, 400, 400, 400)
|
||||
curses.init_color(10, 450, 450, 450)
|
||||
curses.init_color(11, 500, 500, 500)
|
||||
curses.init_color(12, 550, 550, 550)
|
||||
curses.init_color(13, 600, 600, 600)
|
||||
curses.init_color(14, 650, 650, 650)
|
||||
curses.init_color(15, 700, 700, 700)
|
||||
curses.init_color(16, 750, 750, 750)
|
||||
curses.init_color(17, 800, 800, 800)
|
||||
curses.init_color(18, 850, 850, 850)
|
||||
curses.init_color(19, 900, 900, 900)
|
||||
curses.init_color(20, 950, 950, 950)
|
||||
curses.init_pair(1, 0, 1)
|
||||
curses.init_pair(2, 0, 2)
|
||||
curses.init_pair(3, 0, 3)
|
||||
curses.init_pair(4, 0, 4)
|
||||
curses.init_pair(5, 0, 5)
|
||||
curses.init_pair(6, 0, 6)
|
||||
curses.init_pair(7, 0, 7)
|
||||
curses.init_pair(8, 0, 8)
|
||||
curses.init_pair(9, 0, 9)
|
||||
curses.init_pair(10, 0, 10)
|
||||
curses.init_pair(11, 0, 11)
|
||||
curses.init_pair(12, 0, 12)
|
||||
curses.init_pair(13, 0, 13)
|
||||
curses.init_pair(14, 0, 14)
|
||||
curses.init_pair(15, 0, 15)
|
||||
curses.init_pair(16, 0, 16)
|
||||
curses.init_pair(17, 0, 17)
|
||||
curses.init_pair(18, 0, 18)
|
||||
curses.init_pair(19, 0, 19)
|
||||
curses.init_pair(20, 0, 20)
|
||||
curses.init_pair(21, 17, 2)
|
||||
|
||||
|
||||
def new_monochrome():
|
||||
global COLOR_COUNT
|
||||
COLOR_COUNT = 101
|
||||
curses.init_color(1, 0, 0, 0)
|
||||
curses.init_color(2, 10, 10, 10)
|
||||
curses.init_color(3, 20, 20, 20)
|
||||
curses.init_color(4, 30, 30, 30)
|
||||
curses.init_color(5, 40, 40, 40)
|
||||
curses.init_color(6, 50, 50, 50)
|
||||
curses.init_color(7, 60, 60, 60)
|
||||
curses.init_color(8, 70, 70, 70)
|
||||
curses.init_color(9, 80, 80, 80)
|
||||
curses.init_color(10, 90, 90, 90)
|
||||
curses.init_color(11, 100, 100, 100)
|
||||
curses.init_color(12, 110, 110, 110)
|
||||
curses.init_color(13, 120, 120, 120)
|
||||
curses.init_color(14, 130, 130, 130)
|
||||
curses.init_color(15, 140, 140, 140)
|
||||
curses.init_color(16, 150, 150, 150)
|
||||
curses.init_color(17, 160, 160, 160)
|
||||
curses.init_color(18, 170, 170, 170)
|
||||
curses.init_color(19, 180, 180, 180)
|
||||
curses.init_color(20, 190, 190, 190)
|
||||
curses.init_color(21, 200, 200, 200)
|
||||
curses.init_color(22, 210, 210, 210)
|
||||
curses.init_color(23, 220, 220, 220)
|
||||
curses.init_color(24, 230, 230, 230)
|
||||
curses.init_color(25, 240, 240, 240)
|
||||
curses.init_color(26, 250, 250, 250)
|
||||
curses.init_color(27, 260, 260, 260)
|
||||
curses.init_color(28, 270, 270, 270)
|
||||
curses.init_color(29, 280, 280, 280)
|
||||
curses.init_color(30, 290, 290, 290)
|
||||
curses.init_color(31, 300, 300, 300)
|
||||
curses.init_color(32, 310, 310, 310)
|
||||
curses.init_color(33, 320, 320, 320)
|
||||
curses.init_color(34, 330, 330, 330)
|
||||
curses.init_color(35, 340, 340, 340)
|
||||
curses.init_color(36, 350, 350, 350)
|
||||
curses.init_color(37, 360, 360, 360)
|
||||
curses.init_color(38, 370, 370, 370)
|
||||
curses.init_color(39, 380, 380, 380)
|
||||
curses.init_color(40, 390, 390, 390)
|
||||
curses.init_color(41, 400, 400, 400)
|
||||
curses.init_color(42, 410, 410, 410)
|
||||
curses.init_color(43, 420, 420, 420)
|
||||
curses.init_color(44, 430, 430, 430)
|
||||
curses.init_color(45, 440, 440, 440)
|
||||
curses.init_color(46, 450, 450, 450)
|
||||
curses.init_color(47, 460, 460, 460)
|
||||
curses.init_color(48, 470, 470, 470)
|
||||
curses.init_color(49, 480, 480, 480)
|
||||
curses.init_color(50, 490, 490, 490)
|
||||
curses.init_color(51, 500, 500, 500)
|
||||
curses.init_color(52, 510, 510, 510)
|
||||
curses.init_color(53, 520, 520, 520)
|
||||
curses.init_color(54, 530, 530, 530)
|
||||
curses.init_color(55, 540, 540, 540)
|
||||
curses.init_color(56, 550, 550, 550)
|
||||
curses.init_color(57, 560, 560, 560)
|
||||
curses.init_color(58, 570, 570, 570)
|
||||
curses.init_color(59, 580, 580, 580)
|
||||
curses.init_color(60, 590, 590, 590)
|
||||
curses.init_color(61, 600, 600, 600)
|
||||
curses.init_color(62, 610, 610, 610)
|
||||
curses.init_color(63, 620, 620, 620)
|
||||
curses.init_color(64, 630, 630, 630)
|
||||
curses.init_color(65, 640, 640, 640)
|
||||
curses.init_color(66, 650, 650, 650)
|
||||
curses.init_color(67, 660, 660, 660)
|
||||
curses.init_color(68, 670, 670, 670)
|
||||
curses.init_color(69, 680, 680, 680)
|
||||
curses.init_color(70, 690, 690, 690)
|
||||
curses.init_color(71, 700, 700, 700)
|
||||
curses.init_color(72, 710, 710, 710)
|
||||
curses.init_color(73, 720, 720, 720)
|
||||
curses.init_color(74, 730, 730, 730)
|
||||
curses.init_color(75, 740, 740, 740)
|
||||
curses.init_color(76, 750, 750, 750)
|
||||
curses.init_color(77, 760, 760, 760)
|
||||
curses.init_color(78, 770, 770, 770)
|
||||
curses.init_color(79, 780, 780, 780)
|
||||
curses.init_color(80, 790, 790, 790)
|
||||
curses.init_color(81, 800, 800, 800)
|
||||
curses.init_color(82, 810, 810, 810)
|
||||
curses.init_color(83, 820, 820, 820)
|
||||
curses.init_color(84, 830, 830, 830)
|
||||
curses.init_color(85, 840, 840, 840)
|
||||
curses.init_color(86, 850, 850, 850)
|
||||
curses.init_color(87, 860, 860, 860)
|
||||
curses.init_color(88, 870, 870, 870)
|
||||
curses.init_color(89, 880, 880, 880)
|
||||
curses.init_color(90, 890, 890, 890)
|
||||
curses.init_color(91, 900, 900, 900)
|
||||
curses.init_color(92, 910, 910, 910)
|
||||
curses.init_color(93, 920, 920, 920)
|
||||
curses.init_color(94, 930, 930, 930)
|
||||
curses.init_color(95, 940, 940, 940)
|
||||
curses.init_color(96, 950, 950, 950)
|
||||
curses.init_color(97, 960, 960, 960)
|
||||
curses.init_color(98, 970, 970, 970)
|
||||
curses.init_color(99, 980, 980, 980)
|
||||
curses.init_color(100, 990, 990, 990)
|
||||
curses.init_pair(1, 0, 1)
|
||||
curses.init_pair(2, 0, 2)
|
||||
curses.init_pair(3, 0, 3)
|
||||
curses.init_pair(4, 0, 4)
|
||||
curses.init_pair(5, 0, 5)
|
||||
curses.init_pair(6, 0, 6)
|
||||
curses.init_pair(7, 0, 7)
|
||||
curses.init_pair(8, 0, 8)
|
||||
curses.init_pair(9, 0, 9)
|
||||
curses.init_pair(10, 0, 10)
|
||||
curses.init_pair(11, 0, 11)
|
||||
curses.init_pair(12, 0, 12)
|
||||
curses.init_pair(13, 0, 13)
|
||||
curses.init_pair(14, 0, 14)
|
||||
curses.init_pair(15, 0, 15)
|
||||
curses.init_pair(16, 0, 16)
|
||||
curses.init_pair(17, 0, 17)
|
||||
curses.init_pair(18, 0, 18)
|
||||
curses.init_pair(19, 0, 19)
|
||||
curses.init_pair(20, 0, 20)
|
||||
curses.init_pair(21, 0, 21)
|
||||
curses.init_pair(22, 0, 22)
|
||||
curses.init_pair(23, 0, 23)
|
||||
curses.init_pair(24, 0, 24)
|
||||
curses.init_pair(25, 0, 25)
|
||||
curses.init_pair(26, 0, 26)
|
||||
curses.init_pair(27, 0, 27)
|
||||
curses.init_pair(28, 0, 28)
|
||||
curses.init_pair(29, 0, 29)
|
||||
curses.init_pair(30, 0, 30)
|
||||
curses.init_pair(31, 0, 31)
|
||||
curses.init_pair(32, 0, 32)
|
||||
curses.init_pair(33, 0, 33)
|
||||
curses.init_pair(34, 0, 34)
|
||||
curses.init_pair(35, 0, 35)
|
||||
curses.init_pair(36, 0, 36)
|
||||
curses.init_pair(37, 0, 37)
|
||||
curses.init_pair(38, 0, 38)
|
||||
curses.init_pair(39, 0, 39)
|
||||
curses.init_pair(40, 0, 40)
|
||||
curses.init_pair(41, 0, 41)
|
||||
curses.init_pair(42, 0, 42)
|
||||
curses.init_pair(43, 0, 43)
|
||||
curses.init_pair(44, 0, 44)
|
||||
curses.init_pair(45, 0, 45)
|
||||
curses.init_pair(46, 0, 46)
|
||||
curses.init_pair(47, 0, 47)
|
||||
curses.init_pair(48, 0, 48)
|
||||
curses.init_pair(49, 0, 49)
|
||||
curses.init_pair(50, 0, 50)
|
||||
curses.init_pair(51, 0, 51)
|
||||
curses.init_pair(52, 0, 52)
|
||||
curses.init_pair(53, 0, 53)
|
||||
curses.init_pair(54, 0, 54)
|
||||
curses.init_pair(55, 0, 55)
|
||||
curses.init_pair(56, 0, 56)
|
||||
curses.init_pair(57, 0, 57)
|
||||
curses.init_pair(58, 0, 58)
|
||||
curses.init_pair(59, 0, 59)
|
||||
curses.init_pair(60, 0, 60)
|
||||
curses.init_pair(61, 0, 61)
|
||||
curses.init_pair(62, 0, 62)
|
||||
curses.init_pair(63, 0, 63)
|
||||
curses.init_pair(64, 0, 64)
|
||||
curses.init_pair(65, 0, 65)
|
||||
curses.init_pair(66, 0, 66)
|
||||
curses.init_pair(67, 0, 67)
|
||||
curses.init_pair(68, 0, 68)
|
||||
curses.init_pair(69, 0, 69)
|
||||
curses.init_pair(70, 0, 70)
|
||||
curses.init_pair(71, 0, 71)
|
||||
curses.init_pair(72, 0, 72)
|
||||
curses.init_pair(73, 0, 73)
|
||||
curses.init_pair(74, 0, 74)
|
||||
curses.init_pair(75, 0, 75)
|
||||
curses.init_pair(76, 0, 76)
|
||||
curses.init_pair(77, 0, 77)
|
||||
curses.init_pair(78, 0, 78)
|
||||
curses.init_pair(79, 0, 79)
|
||||
curses.init_pair(80, 0, 80)
|
||||
curses.init_pair(81, 0, 81)
|
||||
curses.init_pair(82, 0, 82)
|
||||
curses.init_pair(83, 0, 83)
|
||||
curses.init_pair(84, 0, 84)
|
||||
curses.init_pair(85, 0, 85)
|
||||
curses.init_pair(86, 0, 86)
|
||||
curses.init_pair(87, 0, 87)
|
||||
curses.init_pair(88, 0, 88)
|
||||
curses.init_pair(89, 0, 89)
|
||||
curses.init_pair(90, 0, 90)
|
||||
curses.init_pair(91, 0, 91)
|
||||
curses.init_pair(92, 0, 92)
|
||||
curses.init_pair(93, 0, 93)
|
||||
curses.init_pair(94, 0, 94)
|
||||
curses.init_pair(95, 0, 95)
|
||||
curses.init_pair(96, 0, 96)
|
||||
curses.init_pair(97, 0, 97)
|
||||
curses.init_pair(98, 0, 98)
|
||||
curses.init_pair(99, 0, 99)
|
||||
curses.init_pair(100, 0, 100)
|
||||
curses.init_pair(101, 81, 2)
|
||||
|
||||
|
||||
def main(screen):
|
||||
rows, cols = screen.getmaxyx()
|
||||
rows -= 1
|
||||
# new_monochrome()
|
||||
init_colors()
|
||||
char = 0
|
||||
while True:
|
||||
matrix = calculate_set(cols, rows)
|
||||
print_set(screen, cols, rows, matrix, char)
|
||||
char = screen.getch()
|
||||
if handle_keyboard(char):
|
||||
break
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
curses.wrapper(main)
|
||||
|
||||
Loading…
Reference in New Issue