-
Notifications
You must be signed in to change notification settings - Fork 0
/
dragon2.py
71 lines (60 loc) · 1.53 KB
/
dragon2.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
#SETUP
import pygame, time, sys
#ITERATE
def multiply(number_list,number):
for i in range(len(number_list)):
number_list[i] *= (number)
return number_list
def iterate(turns_list):
negated_flipped_turns_list = multiply(turns_list[::-1],-1)
turns_list.append(1)
turns_list += negated_flipped_turns_list
return turns_list
#TURNS
# north = 0
# east = 1
# south = 2
# west = 3
# right = 1
# left = -1
def turn(cardinal,turn_direction):
return (cardinal + 4 + turn_direction) % 4
def get_coords(cardinal,line_length):
if cardinal % 2 == 0:
return [(cardinal-1)*line_length,0]
else:
return [0,(cardinal-2)*line_length]
#PLOT POINTS
def generate_plot_points(x_initial,y_initial,turns_list,line_length):
plot_points = [[x_initial,y_initial]]
x = x_initial
y = y_initial
cardinal = 0
coords = []
for item in turns_list:
cardinal = turn(cardinal,item)
coords = get_coords(cardinal,line_length)
x += coords[0]
y += coords[1]
plot_points.append([x,y])
return plot_points
#DISPLAY
def display(plot_points):
pygame.init()
screen = pygame.display.set_mode([600,600])
screen.fill([255,255,255])
for i in range(2,len(plot_points)):
pygame.draw.lines(screen,[0,0,0],False,plot_points[:i],1)
pygame.display.flip()
time.sleep(.002)
#EXECUTION
turns_list = []
for i in range(11):
iterate(turns_list)
plot_points = generate_plot_points(200,400,turns_list,8)
display(plot_points)
#EXIT
while True:
for event in pygame.event.get():
if event.type == pygame.QUIT:
sys.exit()