В следующих примерах и пояснениях я буду использовать простое изображение, сгенерированное визуализированным текстом:
font = pygame.font.SysFont('Times New Roman', 50)
text = font.render('image', False, (255, 255, 0))
image = pygame.Surface((text.get_width()+1, text.get_height()+1))
pygame.draw.rect(image, (0, 0, 255), (1, 1, *text.get_size()))
image.blit(text, (1, 1))
Изображение (pygame.Surface) можно повернуть на pygame.transform.rotate.
Если это происходит постепенно в цикле, изображение искажается и быстро увеличивается:
while not done:
# [...]
image = pygame.transform.rotate(image, 1)
screen.blit(image, pos)
pygame.display.flip()
Это является причиной, потому что ограничивающий прямоугольник повернутого изображения всегда больше, чем ограничивающий прямоугольник исходного изображения (за исключением некоторых поворотов, кратных 90 градусам).
Изображение искажается из-заразмножить копии.Каждый поворот генерирует небольшую ошибку (неточность).Сумма ошибок увеличивается, и изображения затухают.
Это можно исправить, сохранив исходное изображение и "блитировав" изображение, которое было создано с помощью одной операции поворота из исходного изображения.
angle = 0
while not done:
# [...]
rotated_image = pygame.transform.rotate(image, angle)
angle += 1
screen.blit(rotated_image, pos)
pygame.display.flip()
Теперь изображение, по-видимому, произвольно меняет свою позицию, поскольку размер изображения изменяется при повороте, а начало координат всегда находится слева вверху от ограничителяпрямоугольник изображения.
Это можно компенсировать, сравнивая ограничивающую рамку с выравниванием по оси изображения до вращения и после вращения.
Для следующей математики pygame.math.Vector2 используется.Обратите внимание на координаты экрана, точки y вниз по экрану, но точки математической оси y образуют нижнюю часть к верхней.Это приводит к тому, что ось y должна быть «перевернута» во время вычислений
Создать список с 4 угловыми точками ограничительной рамки:
w, h = image.get_size()
box = [pygame.math.Vector2(p) for p in [(0, 0), (w, 0), (w, -h), (0, -h)]]
Повернуть векторы к угловым точкампо pygame.math.Vector2.rotate:
box_rotate = [p.rotate(angle) for p in box]
Получите минимум и максимум повернутых точек:
min_box = (min(box_rotate, key=lambda p: p[0])[0], min(box_rotate, key=lambda p: p[1])[1])
max_box = (max(box_rotate, key=lambda p: p[0])[0], max(box_rotate, key=lambda p: p[1])[1])
Рассчитайте «компенсированный» источник верхнего левого угла.точка изображения, добавив минимум повернутой рамки в положение.Для координаты y max_box[1] является минимумом из-за «переворачивания» вдоль оси y:
origin = (pos[0] + min_box[0], pos[1] - max_box[1])
rotated_image = pygame.transform.rotate(image, angle)
screen.blit(rotated_image, origin)
Чтобы определить точку поворотана исходном изображении должен быть рассчитан «перевод» оси относительно верхнего левого угла изображения, а положение «блиц» изображения должно быть смещено переводом.
Определитецентральная точка, например, в центре изображения:
pivot = pygame.math.Vector2(w/2, -h/2)
Рассчитать перевод повернутого центра:
pivot_rotate = pivot.rotate(angle)
pivot_move = pivot_rotate - pivot
Окончательно рассчитать происхождение повернутого изображения:
origin = (pos[0] + min_box[0] - pivot_move[0], pos[1] - max_box[1] + pivot_move[1])
rotated_image = pygame.transform.rotate(image, angle)
screen.blit(rotated_image, origin)
Если необходимо дополнительно увеличить изображение, то при расчете источника изображения необходимо учитывать увеличение:
move = (-pivot[0] + min_box[0] - pivot_move[0], pivot[1] - max_box[1] + pivot_move[1])
origin = (pos[0] + zoom * move[0], pos[1] + zoom * move[1])
rotozoom_image = pygame.transform.rotozoom(image, angle, zoom)
screen.blit(rotozoom_image, origin)
В следующем примере программы функция blitRotate выполняет все вышеописанные шаги и «перетягивает» повернутое изображение на поверхность.pos - позиция изображения.originPos - это точка на изображении, расположенная на pos, и точка поворота:
import pygame
import pygame.font
pygame.init()
size = (400,400)
screen = pygame.display.set_mode(size)
clock = pygame.time.Clock()
def blitRotate(surf, image, pos, originPos, angle, zoom):
# calcaulate the axis aligned bounding box of the rotated image
w, h = image.get_size()
box = [pygame.math.Vector2(p) for p in [(0, 0), (w, 0), (w, -h), (0, -h)]]
box_rotate = [p.rotate(angle) for p in box]
min_box = (min(box_rotate, key=lambda p: p[0])[0], min(box_rotate, key=lambda p: p[1])[1])
max_box = (max(box_rotate, key=lambda p: p[0])[0], max(box_rotate, key=lambda p: p[1])[1])
# calculate the translation of the pivot
pivot = pygame.math.Vector2(originPos[0], -originPos[1])
pivot_rotate = pivot.rotate(angle)
pivot_move = pivot_rotate - pivot
# calculate the upper left origin of the rotated image
move = (-originPos[0] + min_box[0] - pivot_move[0], -originPos[1] - max_box[1] + pivot_move[1])
origin = (pos[0] + zoom * move[0], pos[1] + zoom * move[1])
# get a rotated image
rotozoom_image = pygame.transform.rotozoom(image, angle, zoom)
# rotate and blit the image
surf.blit(rotozoom_image, origin)
# draw rectangle around the image
pygame.draw.rect (surf, (255, 0, 0), (*origin, *rotozoom_image.get_size()),2)
font = pygame.font.SysFont('Times New Roman', 50)
text = font.render('image', False, (255, 255, 0))
image = pygame.Surface((text.get_width()+1, text.get_height()+1))
pygame.draw.rect(image, (0, 0, 255), (1, 1, *text.get_size()))
image.blit(text, (1, 1))
w, h = image.get_size()
angle = 0
zoom = 1.0
done = False
while not done:
clock.tick(60)
for event in pygame.event.get():
if event.type == pygame.QUIT:
done = True
elif event.type == pygame.KEYDOWN:
if event.key==pygame.K_ESCAPE:
done = True
pos = (screen.get_width()/2, screen.get_height()/2)
pos = (200, 200)
screen.fill(0)
blitRotate(screen, image, pos, (w/2, h/2), angle, zoom)
angle += 1
zoom += 0.01
pygame.draw.line(screen, (0, 255, 0), (pos[0]-20, pos[1]), (pos[0]+20, pos[1]), 3)
pygame.draw.line(screen, (0, 255, 0), (pos[0], pos[1]-20), (pos[0], pos[1]+20), 3)
pygame.draw.circle(screen, (0, 255, 0), pos, 7, 0)
pygame.display.flip()
pygame.quit()