/
bird.py
87 lines (69 loc) · 2.31 KB
/
bird.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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
# imports
import pygame
import pygame.transform as transform
import pygame.image as image
import pygame.mask as mask
import os
# local imports
BIRD_IMGS = [transform.scale2x(image.load(os.path.join("imgs", "bird1.png"))), transform.scale2x(image.load(
os.path.join("imgs", "bird2.png"))), transform.scale2x(image.load(os.path.join("imgs", "bird3.png")))]
class Bird:
IMGS = BIRD_IMGS
MAX_ROTATION = 25
ROT_VEL = 20
ANIMATION_TIME = 5
def __init__(self, x, y):
self.x = x
self.y = y
self.tilt = 0
self.tick_count = 0
self.vel = 0
self.height = self.y
self.img_count = 0
self.img = self.IMGS[0]
def jump(self):
self.vel = -10.5
self.tick_count = 0
self.height = self.y
def move(self):
self.tick_count += 1
# displacement calculation
d = self.vel * self.tick_count + 1.5 * self.tick_count ** 2
if d >= 16:
d = 16
if d < 0:
d -= 2
self.y = self.y + d
# tilt calculation
if d < 0 or self.y < self.height + 50:
if self.tilt < self.MAX_ROTATION:
self.tilt = self.MAX_ROTATION
else:
if self.tilt > -90:
self.tilt -= self.ROT_VEL
def draw(self, win, n):
p = n / 20
self.img.set_alpha(255 - (200 * p))
self.img_count += 1
# image logic
if self.img_count < self.ANIMATION_TIME:
self.img = self.IMGS[0]
elif self.img_count < self.ANIMATION_TIME * 2:
self.img = self.IMGS[1]
elif self.img_count < self.ANIMATION_TIME * 3:
self.img = self.IMGS[2]
elif self.img_count < self.ANIMATION_TIME * 4:
self.img = self.IMGS[1]
elif self.img_count == self.ANIMATION_TIME * 4 + 1:
self.img = self.IMGS[0]
self.img_count = 0
if self.tilt <= -80:
self.img = self.IMGS[1]
self.img_count = self.ANIMATION_TIME * 2
# image rotation
rotated_img = transform.rotate(self.img, self.tilt)
new_rect = rotated_img.get_rect(
center=self.img.get_rect(topleft=(self.x, self.y)).center)
win.blit(rotated_img, new_rect.topleft)
def get_mask(self):
return mask.from_surface(self.img)