前言：上次，参考了Ahab的文章，实现了一个简单的降雪脚本。但是，身为一个强迫症患者，怎么能用圆代替大自然那完美的六边形的雪花那。所以，今天，我就在上次的基础上，改进了一下,对比一下:
1.0版本
2.0版本
是不是和大自然的雪花更加像了。
好，接下来，看一下如何实现。
源码：
'''
人工降雪v2.0
Data:2018-12-11
Author:Lingyin
reference:https://mp.weixin.qq.com/s/fki-eGBzrSsz2xDbk--vpw
'''
import pygame
import random
import os
from pygame.locals import *
from pygame.compat import geterror
if not pygame.mixer: print('Warning, sound disabled')
SIZE = (1300,700)
#设置一些基本的颜色
BLACK = (0,0,0)
WHITE = (255,255,255)
RED = (255,0,0)
GREEN = (0,255,0)
BLUE = (0,0,255)
main_dir = os.path.split(os.path.abspath(__file__))[0]
data_dir = os.path.join(main_dir, 'data')
#初始化
pygame.init()
screen = pygame.display.set_mode(SIZE)
#设置标题
pygame.display.set_caption('唯美雪景')
#设置鼠标光标不可见
pygame.mouse.set_visible(0)
#加载图片，利用os模块，实现跨平台
def load_image(name):
fullname = os.path.join(data_dir,name)
try:
image = pygame.image.load(fullname)
except pygame.error as e:
raise e
image = image.convert()
return image
def load_sound(name):
class NoneSound:
def play(self): pass
if not pygame.mixer or not pygame.mixer.get_init():
return NoneSound()
fullname = os.path.join(data_dir, name)
print(fullname)
try:
pygame.mixer.music.load(fullname)
pygame.mixer.music.play()
except pygame.error:
print('Cannot load sound: %s' % fullname)
raise SystemExit(str(geterror()))
class Snow(pygame.sprite.Sprite):
def __init__(self):
pygame.sprite.Sprite.__init__(self)
self.width = random.randrange(5,15)
self.image = pygame.transform.scale(load_image('snowflake.png'),(self.width,self.width))
self.image.set_colorkey(BLACK)
self.rect = self.image.get_rect()
self.rect.x = random.randrange(0,1300)
self.rect.y = random.randrange(0,700)
self.speedx = random.randrange(-3,5)
self.speedy = random.randrange(3,5)
def update(self):
self.rect.x += self.speedx
self.rect.y += self.speedy
if self.rect.x > SIZE[0] or self.rect.x < 0 or self.rect.y > SIZE[1]:
self.rect.x = random.randrange(0,1300)
self.rect.y = 0
flow = pygame.sprite.Group()
for i in range(300):
s = Snow()
flow.add(s)
def main():
#加载背景
background = load_image('Lingyin0.jpg')
screen.blit(background,(0,0))
#显示背景
pygame.display.flip()
clock = pygame.time.Clock()
snow = Snow()
load_sound('flower.mp3')
#主循环
going = True
while going:
clock.tick(20)
for event in pygame.event.get():
if event.type == QUIT:
going = False
elif event.type == KEYDOWN and event.key == K_ESCAPE:
going = False
flow.update()
flow.draw(screen)
pygame.display.flip()
screen.blit(background, (0, 0))
pygame.quit()
if __name__ == '__main__':
main()
其实和上次的代码，没有太大的区别。最主要的就是，这次我增加了一个Snow类。下面，我逐步讲解。
class Snow(pygame.sprite.Sprite):
def __init__(self):
pygame.sprite.Sprite.__init__(self)
self.width = random.randrange(5,15)
self.image = pygame.transform.scale(load_image('snowflake.png'),(self.width,self.width))
self.image.set_colorkey(BLACK)
self.rect = self.image.get_rect()
self.rect.x = random.randrange(0,1300)
self.rect.y = random.randrange(0,700)
self.speedx = random.randrange(-3,5)
self.speedy = random.randrange(3,5)
def update(self):
self.rect.x += self.speedx
self.rect.y += self.speedy
if self.rect.x > SIZE[0] or self.rect.x < 0 or self.rect.y > SIZE[1]:
self.rect.x = random.randrange(0,1300)
self.rect.y = 0#random.randrange(0,700)
首先，定义了一个Snow类，它继承自pygame.sprite.Sprite,这是pygame模块内部的一个类，不用管它是什么，只管用就行。
load_image('snowflake.png')
这个函数和1.0版本的一样，不需要讲解。
下面我们看看pygame.transform.scale()这个函数
self.image = pygame.transform.scale(load_image('snowflake.png'),(self.width,self.width))
这个函数的作用就是改变图片的尺寸，所以它接受两个参数，一个是我们加载的图片，另一个是元组，也就是要改变的大小。将改变后的图片交给自定义的self.image.由于雪花的大小并不是一样的，所以在实例初始化的时候，我们随机生成了它的大小:
self.width = random.randrange(5,15)
用过贴图做程序的同学一定知道，我们加载的图片，是一个矩形，但是当我们玩游戏的时候，角色并不是矩形的啊？这就需要下面这条语句来处理了:
self.image.set_colorkey(BLACK)
用它处理后，我们的雪花，看起来才是一个六边形的形状，那如果，不加这条语句会是什么效果，看一下:
看见雪花周围的黑色框框了吗
self.rect = self.image.get_rect()
self.rect.x = random.randrange(0,1300)
self.rect.y = random.randrange(0,700)
self.speedx = random.randrange(-3,5)
self.speedy = random.randrange(3,5)
然后，通过self.image的get_rect方法，获取了雪花图片本身这个矩形
并且通过随机函数对它的起始坐标进行了赋值。最后，对它在x和y轴的速度进行了随机赋值。
接下来是update()函数
def update(self):
self.rect.x += self.speedx
self.rect.y += self.speedy
if self.rect.x > SIZE[0] or self.rect.x < 0 or self.rect.y > SIZE[1]:
self.rect.x = random.randrange(0,1300)
self.rect.y = 0
update()是用来移动雪花，也就是更新雪花的位置的。雪花的x和y分别加上各自的速度。最后判断，如果雪花移出了屏幕范围，再将它的x随机赋值，y赋值为0，即从屏幕最上方下落。
flow = pygame.sprite.Group()
pygame.sprite模块有一个Group类，大致应用就是可以将我们定义的类的实例包含进来，然后，同时进行一个操作。具体的使用方法，可以去pygame的官网去查阅
for i in range(300):
s = Snow()
flow.add(s)
flow有一个add方法，将实例添加进去。这里，我通过迭代，创建了300个雪花的实例，如果你想让美女体验一场暴雪的话，可以将它的值调到更大，但是，启动速度可就降下来了！！！
flow.update()
flow.draw(screen)
pygame.display.flip()
screen.blit(background, (0, 0))
最后，将所有的雪花实例update()，即更新位置，然后画到屏幕上->显示->擦除->update()->画->....一直循环。
源码和图片放到了我的GitHub上:
地址：https://github.com/dmzlingyin/pygame