pygame 笔记-9 图片旋转及边界反弹

h5或flash中,可以直接对矢量对象,比如line, rectange旋转,但是pygame中,仅支持对image旋转,本以为这个是很简单的事情,但是发现还是有很多小猫腻的,记录一下:

先看一个错误的版本:

 import pygame
import sys pygame.init() SIZE = WIDTH, HEIGHT = 200, 400
BLACK = 0, 0, 0
angle = 1 screen = pygame.display.set_mode(SIZE)
leaf = pygame.image.load("leaf.png")
leafRect = leaf.get_rect()
# 定位到舞台中心
leafRect = leafRect.move((WIDTH - leafRect.width) / 2, (HEIGHT - leafRect.height) / 2) clock = pygame.time.Clock() while True: for event in pygame.event.get():
if event.type == pygame.QUIT:
sys.exit() # 旋转图片
leaf = pygame.transform.rotate(leaf, angle)
angle += 1 # 默认背景为白色,所以每渲染一帧,要对背景重新填充,否则会有上一帧的残影
screen.fill(BLACK)
# 将旋转后的图象,渲染矩形里
screen.blit(leaf, leafRect)
# 正式渲染
pygame.display.update()
# 控制帧数<=100
clock.tick(100)

leaf-01

代码中的leaf.png图例如下:

pygame 笔记-9 图片旋转及边界反弹

跑一把:

pygame 笔记-9 图片旋转及边界反弹

这明显跟我想的不一样!代码里并没有对叶子做移动操作,只是每帧旋转1度而已,为啥它要飘到舞台之外?

仔细review了下代码,25行:leaf = pygame.transform.rotate(leaf, angle) 这里有问题,pygame在这方面做得说实话不算太好,字面上的意思,这行的效果,应该是每次在原来的基础上,继续加速旋转(因为每次angle+1,相当于每帧旋转角度越来越大)。

需要用一个新变量存储旋转后的图片,参考下面的代码:

 import pygame
import sys pygame.init() SIZE = WIDTH, HEIGHT = 200, 400
BLACK = 0, 0, 0
angle = 1 screen = pygame.display.set_mode(SIZE)
leaf = pygame.image.load("leaf.png")
leafRect = leaf.get_rect()
leafRect = leafRect.move((WIDTH - leafRect.width) / 2, (HEIGHT - leafRect.height) / 2) clock = pygame.time.Clock() while True: for event in pygame.event.get():
if event.type == pygame.QUIT:
sys.exit() keys = pygame.key.get_pressed() # 旋转图片(注意:这里要搞一个新变量,存储旋转后的图片)
newLeaf = pygame.transform.rotate(leaf, angle)
angle += 1 screen.fill(BLACK)
screen.blit(newLeaf, leafRect)
pygame.display.update()
clock.tick(100)

leaf-02

总算不乱跑,开始正经的旋转了: 

pygame 笔记-9 图片旋转及边界反弹

但是仔细观察,还是有点小问题,旋转过程中,叶子的中心位置总在晃动,预期效果最好是旋转过程中,中心点不变。至于晃动的原因,叶子图片并不是一个圆形,pygame中,任何一个Surface对象,总有一个外切的矩形对象(通过get_rect()方法可以获得),图片旋转后,这个外切Rect对象的尺寸跟着变化,导致中心点也变化了。

校正方法如下(为了方便观察,我们把叶子外围正切的矩形也画出来)

做为对比,刚才的中心晃动版本,也把矩形画出来,参考以下代码:

 import pygame
import sys pygame.init() SIZE = WIDTH, HEIGHT = 200, 400
BLACK = 0, 0, 0
angle = 1 screen = pygame.display.set_mode(SIZE)
leaf = pygame.image.load("leaf.png")
leafRect = leaf.get_rect()
leafRect = leafRect.move((WIDTH - leafRect.width) / 2, (HEIGHT - leafRect.height) / 2) clock = pygame.time.Clock() while True: for event in pygame.event.get():
if event.type == pygame.QUIT:
sys.exit() keys = pygame.key.get_pressed() # 旋转图片(注意:这里要搞一个新变量,存储旋转后的图片)
newLeaf = pygame.transform.rotate(leaf, angle)
newRect = newLeaf.get_rect()
angle += 1 screen.fill(BLACK)
screen.blit(newLeaf, leafRect)
pygame.draw.rect(screen, (255, 0, 0), leafRect, 1)
pygame.draw.rect(screen, (0, 255, 0), newRect, 1)
pygame.display.update()
clock.tick(100)

leaf-02-with-rect

校正中心点后的版本:

 import pygame
import sys pygame.init() SIZE = WIDTH, HEIGHT = 200, 400
BLACK = 0, 0, 0
angle = 1 screen = pygame.display.set_mode(SIZE)
leaf = pygame.image.load("leaf.png")
leafRect = leaf.get_rect()
leafRect = leafRect.move((WIDTH - leafRect.width) / 2, (HEIGHT - leafRect.height) / 2) clock = pygame.time.Clock() while True: for event in pygame.event.get():
if event.type == pygame.QUIT:
sys.exit() keys = pygame.key.get_pressed() # 旋转图片(注意:这里要搞一个新变量,存储旋转后的图片)
newLeaf = pygame.transform.rotate(leaf, angle)
# 校正旋转图片的中心点
newRect = newLeaf.get_rect(center=leafRect.center)
angle += 1 screen.fill(BLACK)
# 这里要用newRect区域,绘制图象
screen.blit(newLeaf, newRect)
pygame.draw.rect(screen, (255, 0, 0), leafRect, 1)
pygame.draw.rect(screen, (0, 255, 0), newRect, 1)
pygame.display.update()
clock.tick(100)

leaf-03

运行后的对照图如下(左侧为中心晃动的版本,右侧为中心点不变的版本)

pygame 笔记-9 图片旋转及边界反弹

思考一下:为什么左侧的图,绿色的矩形框,一直在左上角,而右侧的绿矩形框,会在中心?

答案:Rect对象默认生成时,其left,top属性都是0, 所以旋转后的新图片,其外切矩形一直是在(0,0)位置,但是校正后的版本,get_rect(center=...)这里指定了中心点,所以newRect对象被对齐到跟红色Rect一样的中心位置。

(题外话:至于需要不需要中心点校正,完全看游戏场景,就本例而言,如果只是模拟一片树叶落下,好象中心点晃动,也并不影响视觉感受,所以下面的示例,均没有做中心点校正处理)

结合之前学到的东西,再加点趣味性,让叶子在舞台上飘动起来,同时加入边界碰撞检测,但是要注意:叶子在旋转过程中,外要的矩形尺寸,也会随之变化(这会影响边界检测),为了观察方便,在叶子外面画一个框框。

 import pygame
import sys pygame.init() SIZE = WIDTH, HEIGHT = 200, 400
BLACK = 0, 0, 0
RED = 255, 0, 0
SPEED = [1, 1]
angle = 1 screen = pygame.display.set_mode(SIZE)
originLeaf = pygame.image.load("leaf.png")
originRect = originLeaf.get_rect() clock = pygame.time.Clock() while True: for event in pygame.event.get():
if event.type == pygame.QUIT:
sys.exit() # 利用矩形对象的move方法,让rect移动
originRect = originRect.move(SPEED[0], SPEED[1])
# 注意:这里一定要用一个新变量newLeaf,保存旋转后的image对象
newLeaf = pygame.transform.rotate(originLeaf, angle)
angle += 1 # 注意:这里要定义一个新rect对象,因为图象旋转后,其外切的矩形尺寸会变化
newRect = newLeaf.get_rect()
# 默认的newRect位置在(0,0),要实现矩形外框跟随,必须赋值到新位置
newRect.left, newRect.top = originRect.left, originRect.top # 左右边界反弹的处理
if newRect.left <= 0 or newRect.right >= WIDTH:
SPEED[0] = -SPEED[0] # 上下边界反弹的处理
if newRect.top <= 0 or newRect.bottom >= HEIGHT:
SPEED[1] = -SPEED[1] # 默认背景为白色,所以每渲染一帧,要对背景重新填充,否则会有上一帧的残影
screen.fill(BLACK)
# 画新矩形
pygame.draw.rect(screen, RED, newRect, 1)
# 将旋转后的图象,渲染到新矩形里
screen.blit(newLeaf, originRect)
# 正式渲染
pygame.display.update()
# 控制帧数<=100
clock.tick(100)

基本达到效果了,但是细心观察的话,发现右边界和下边界,碰撞检测其实不够完美,从视觉上看,明明已经到了边界,但是没有及时反弹。

pygame 笔记-9 图片旋转及边界反弹

修正一下:

 import pygame
import sys pygame.init() SIZE = WIDTH, HEIGHT = 200, 400
BLACK = 0, 0, 0
RED = 255, 0, 0
SPEED = [1, 1]
angle = 1 screen = pygame.display.set_mode(SIZE)
originLeaf = pygame.image.load("leaf.png")
originRect = originLeaf.get_rect() clock = pygame.time.Clock() while True: for event in pygame.event.get():
if event.type == pygame.QUIT:
sys.exit() # 利用矩形对象的move方法,让rect移动
originRect = originRect.move(SPEED[0], SPEED[1])
# 注意:这里一定要用一个新变量newLeaf,保存旋转后的image对象
newLeaf = pygame.transform.rotate(originLeaf, angle)
angle += 1 # 注意:这里要定义一个新rect对象,因为图象旋转后,其外切的矩形尺寸会变化
newRect = newLeaf.get_rect()
# 默认的newRect位置在(0,0),要实现矩形外框跟随,必须赋值到新位置
newRect.left, newRect.top = originRect.left, originRect.top # 左右边界反弹的处理
if newRect.left <= 0 or newRect.right >= WIDTH:
SPEED[0] = -SPEED[0]
# 图片移动到接近右边界时(比如:right=198),由于旋转的作用,可能导致叶子一下横过来了,
# right突然会变成210,这样就算速度取反了,由于SPEED[0]=-1,需要10帧后,才能从视觉上真正看到反弹成功(即:210减到200,需要10次)
if newRect.right > WIDTH:
originRect.left = WIDTH - newRect.width # 上下边界反弹的处理
if newRect.top <= 0 or newRect.bottom >= HEIGHT:
SPEED[1] = -SPEED[1]
# 类似右边界的校正处理,防止叶子接近下边界时,由于旋转,一下从横到竖,高度突然加大,导致越界
if newRect.bottom > HEIGHT:
originRect.top = HEIGHT - newRect.height # 默认背景为白色,所以每渲染一帧,要对背景重新填充,否则会有上一帧的残影
screen.fill(BLACK)
# 画新矩形
pygame.draw.rect(screen, RED, newRect, 1)
# 将旋转后的图象,渲染到新矩形里
screen.blit(newLeaf, originRect)
# 正式渲染
pygame.display.update()
# 控制帧数<=100
clock.tick(100)

看上去好多了

pygame 笔记-9 图片旋转及边界反弹

最后再加点料,根据叶子的运动情况,动态调整背景颜色:

 import pygame
import sys
import random
import math pygame.init() SIZE = WIDTH, HEIGHT = 200, 400
BACKGROUND_COLOR = (0, 0, 0) screen = pygame.display.set_mode(SIZE)
leaves = [] class Leaf(object):
def __init__(self, pos=[10, 10], speed=[1, 1]):
self.imageSrc = pygame.image.load("leaf.png")
self.rect = self.imageSrc.get_rect()
self.image = self.imageSrc
self.speed = speed
self.angle = 0
self.pos = pos
self.rect = self.rect.move(pos[0], pos[1]) def move(self):
self.rect = self.rect.move(self.speed[0], self.speed[1])
new_rect = self.image.get_rect()
new_rect.left, new_rect.top = self.rect.left, self.rect.top
if new_rect.left < 0 or new_rect.right > WIDTH:
self.speed[0] = -self.speed[0]
if new_rect.right > WIDTH:
self.rect.left = WIDTH - new_rect.width
if new_rect.left < 0:
self.rect.left = 0
if new_rect.top < 0 or new_rect.bottom > HEIGHT:
self.speed[1] = -self.speed[1]
if new_rect.bottom > HEIGHT:
self.rect.top = HEIGHT - new_rect.height def draw(self):
screen.blit(self.image, self.rect) def rotate(self):
self.image = pygame.transform.rotate(self.imageSrc, self.angle)
self.angle += random.randint(1, 5)
if math.fabs(self.angle) == 360:
self.angle = 0 def init():
for i in range(0, 3):
leaf = Leaf([random.randint(50, WIDTH - 50), random.randint(30, HEIGHT - 200)],
[random.randint(1, 2), random.randint(1, 2)])
leaf.move()
leaves.append(leaf) def to255(x):
if x > 1:
x = 1
return int(255 * math.fabs(x)) clock = pygame.time.Clock()
init() while True: for event in pygame.event.get():
if event.type == pygame.QUIT:
sys.exit() first_rect = leaves[0].rect # 根据第1片叶子的运动情况,随机切换背景色
color_r = to255(first_rect.top / HEIGHT)
color_g = to255(first_rect.left / WIDTH)
color_b = to255(math.fabs(first_rect.left) / (math.fabs(first_rect.top) + math.fabs(first_rect.left) + 1))
# print(color_r, color_g, color_b)
screen.fill((color_r, color_g, color_b)) # 将旋转后的图象,渲染到新矩形里
for item in leaves:
item.rotate()
item.move()
item.draw() # 正式渲染
pygame.display.update()
# 控制帧数<=100
clock.tick(100)

pygame 笔记-9 图片旋转及边界反弹

下边界检测时,还能做些变化,比如:叶子落到地面以下,就重新放回顶端,这样就有漫天落叶,绵绵不断的感觉:

 import pygame
import sys
import random
import math pygame.init() SIZE = WIDTH, HEIGHT = 200, 400
BACKGROUND_COLOR = (230, 255, 230) screen = pygame.display.set_mode(SIZE)
leaves = [] class Leaf(object):
def __init__(self, pos=[10, 10], speed=[1, 1]):
self.imageSrc = pygame.image.load("leaf.png")
self.rect = self.imageSrc.get_rect()
self.image = self.imageSrc
self.speed = speed
self.angle = 0
self.pos = pos
self.rect = self.rect.move(pos[0], pos[1]) def move(self):
self.rect = self.rect.move(self.speed[0], self.speed[1])
new_rect = self.image.get_rect()
new_rect.left, new_rect.top = self.rect.left, self.rect.top
if new_rect.left < 0 or new_rect.right > WIDTH:
self.speed[0] = -self.speed[0]
if new_rect.right > WIDTH:
self.rect.left = WIDTH - new_rect.width
if new_rect.left < 0:
self.rect.left = 0
if new_rect.top > HEIGHT:
self.rect.bottom = 0 def draw(self):
screen.blit(self.image, self.rect) def rotate(self):
self.image = pygame.transform.rotate(self.imageSrc, self.angle)
self.angle += random.randint(1, 5)
if math.fabs(self.angle) == 360:
self.angle = 0 def init():
for i in range(0, 5):
leaf = Leaf([random.randint(50, WIDTH - 50), random.randint(30, HEIGHT)],
[random.randint(1, 2), random.randint(1, 2)])
leaf.move()
leaves.append(leaf) clock = pygame.time.Clock()
init() while True: for event in pygame.event.get():
if event.type == pygame.QUIT:
sys.exit() screen.fill(BACKGROUND_COLOR) # 将旋转后的图象,渲染到新矩形里
for item in leaves:
item.rotate()
item.move()
item.draw() # 正式渲染
pygame.display.update()
# 控制帧数<=100
clock.tick(100)

pygame 笔记-9 图片旋转及边界反弹

参考:https://www.pygame.org/wiki/RotateCenter?parent=CookBook

上一篇:About me & OI这一年


下一篇:Beauty of Array ZOJ - 3872(思维题)