python人工降雪 2.0版本

前言：上次，参考了Ahab的文章，实现了一个简单的降雪脚本。但是，身为一个强迫症患者，怎么能用圆代替大自然那完美的六边形的雪花那。所以，今天，我就在上次的基础上，改进了一下,对比一下:

1.0版本

01.png

2.0版本

00.png

是不是和大自然的雪花更加像了。

好，接下来，看一下如何实现。

源码：

    '''
人工降雪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)

用它处理后，我们的雪花，看起来才是一个六边形的形状，那如果，不加这条语句会是什么效果，看一下:

02.png

看见雪花周围的黑色框框了吗

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