1 files changed, 94 insertions, 0 deletions
diff --git a/chaos_game/sierpinski_triangle.py b/chaos_game/sierpinski_triangle.py
new file mode 100644
index 0000000..4509fac
--- /dev/null
+++ b/chaos_game/sierpinski_triangle.py
@@ -0,0 +1,94 @@
+from pygame.locals import *
+from pygame import Vector2, Surface
+import pygame
+
+import random
+from sys import argv
+
+
+def convert2int(t):
+    return list(map(int, t))
+
+
+def new_point_triangle():
+    attractor_point = random.choice(vertices)
+    path = attractor_point - points[-1]
+    _ = points[-1] + path.normalize() * path.length() / 2
+    points.append(_)
+    return _
+
+
+def generate_surface(iterations):
+    surf = Surface((WIDTH, HEIGHT))
+
+    for _ in range(iterations):
+        new_point = new_point_triangle()
+        surf.set_at(convert2int(new_point), WHITE)
+
+    return surf
+
+
+pygame.init()
+
+WIDTH, HEIGHT = SIZE = 1000, 1000
+
+screen = pygame.display.set_mode(SIZE)
+
+BLACK = Color('black')
+WHITE = Color('white')
+GREEN = Color('green')
+
+font = pygame.font.SysFont('courier', 30)
+
+vertices = [
+    Vector2(200, 200),
+    Vector2(505, 505),
+    Vector2(30, 400)
+]
+
+
+points = [Vector2(201, 200)]
+
+iterations = 0
+static = False
+surface = None
+
+try:
+    iterations = int(argv[1])
+    static = True
+    surface = generate_surface(iterations)
+
+except (ValueError, IndexError):
+    pass
+
+
+while True:
+    for event in pygame.event.get():
+        if event.type == pygame.QUIT:
+            pygame.quit()
+            quit()
+
+    if static:
+        screen.fill(BLACK)
+
+        screen.blit(surface, (0, 0))
+
+        for point in vertices:
+            pygame.draw.circle(screen, GREEN, convert2int(point), 3)
+
+    else:
+        for point in points:
+            screen.set_at(convert2int(point), WHITE)
+
+        for point in vertices:
+            pygame.draw.circle(screen, GREEN, convert2int(point), 3)
+
+        new_point_triangle()
+
+        iterations += 1
+
+    text = font.render('interation: {}'.format(iterations), 4, WHITE)
+    screen.fill(BLACK, text.get_rect(x=10, y=10))
+    screen.blit(text, text.get_rect(x=10, y=10))
+
+    pygame.display.flip()