-
Notifications
You must be signed in to change notification settings - Fork 3
/
snow.py
95 lines (82 loc) · 3.4 KB
/
snow.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
88
89
90
91
92
93
94
95
import random
import sys
import pygame
import snowflake
from snowflake import Snowflake
from pygame.locals import * #@UnusedWildImport
class Game:
def __init__(self, width, height, caption="Game"):
# plethora of fields
self.width = width
self.height = height
self.caption = caption
self.framerate = 60 # FPS
self.foreground_color = (255, 255, 255)
self.background_color = (21, 26, 79)
self.snowflakes = []
self.snowflake_counter = 0
self.snowflake_frequency = 5
self.snowflake_size = 2
self.snowflake_line = [height] * width # for collision detection
self.wind_chance = 1
self.wind_strength = 200
self.show_text = True
# and we're off!
self.initialize()
self.loop()
def initialize(self):
pygame.init()
pygame.display.set_caption(self.caption)
self.screen = pygame.display.set_mode((self.width, self.height))
self.font = pygame.font.Font("fonts/visitor1.ttf", 20)
def loop(self):
self.clock = pygame.time.Clock()
while 1:
gametime = self.clock.get_time()
self.update(gametime)
self.render(gametime)
self.clock.tick(self.framerate)
def update(self, gametime):
# do we need to add more snow?
if self.snowflake_counter > self.snowflake_frequency:
self.snowflake_counter = 0
snowflake = Snowflake(len(self.snowflakes), self.width, self.height, self.snowflake_size, self.foreground_color)
self.snowflakes.append(snowflake)
else:
self.snowflake_counter += 1
# what about some wind?
w_chance = random.randint(0, 100)
w_strength = 0
if w_chance <= self.wind_chance:
w_strength = random.randint(-self.wind_strength, self.wind_strength)
# let it snow, let it snow, let it snow
for snowflake in self.snowflakes:
if snowflake.enabled:
if w_strength != 0:
snowflake.wind = w_strength
snowflake.update(gametime, self.snowflake_line)
# update the other rubbish
if self.show_text:
self.fps_text = self.font.render("FPS: %d" % self.clock.get_fps(), 1, self.foreground_color)
self.snowflake_text = self.font.render("Snowflakes: %d" % len(self.snowflakes), 1, self.foreground_color)
self.handle_input(pygame.event.get())
def render(self, gametime):
surface = pygame.Surface(self.screen.get_size())
surface.convert()
surface.fill(self.background_color)
for snowflake in self.snowflakes:
snowflake.draw(surface)
if self.show_text:
surface.blit(self.fps_text, (8, 6))
surface.blit(self.snowflake_text, (8, 26))
self.screen.blit(surface, (0, 0))
pygame.display.flip()
def handle_input(self, events):
for event in events:
if event.type == pygame.QUIT:
sys.exit()
if event.type == pygame.KEYUP:
if event.key == pygame.K_d:
self.show_text = False if self.show_text else True
if __name__ == "__main__":
game = Game(800, 600, "Snow: a sprinkling of test flakes")