如何在pygame中将三角形旋转到特定角度？

Question

我需要在屏幕中心旋转一个三角形（不是图片）。我看到其他人回答了这个问题，但是三角形不能朝上。

我尝试使用其他人的函数，但它们似乎只部分起作用，就像我上面提到的函数。

import pygame
disp=pygame.display.set_mode((200,200))
import math
def rotate_triange(mouse_pos,triangle_pos):
    #The code here
import time
while True:
    time.sleep(1)
    pygame.Surface.fill(disp,(255,255,255))
    center = (100,100)
    radius = 10
    mouse_position = pygame.mouse.get_pos()
    for event in pygame.event.get():
            pass 
    points = rotate_triangle((100,100),mouse_position)
    pygame.draw.polygon(disp,(0,0,0),points)
    pygame.display.update()

- alex Ilves

---

可能是pygame旋转一条线的重复问题。 - norok2

@norok2 这不是重复的问题。该问题并没有询问如何从向量中获取角度。 - Rabbid76

Answer 1

在pygame中，二维向量算术是通过pygame.math.Vector2实现的。

为鼠标位置和三角形中心定义一个Vector2对象。计算从中心点到鼠标位置的向量的角度(.angle_to())：

vMouse  = pygame.math.Vector2(mouse_pos)
vCenter = pygame.math.Vector2(center)
angle   = pygame.math.Vector2().angle_to(vMouse - vCenter)

定义三角形的三个点围绕(0, 0)并通过角度(.rotate())进行旋转。

points = [(-0.5, -0.866), (-0.5, 0.866), (2.0, 0.0)]
rotated_point = [pygame.math.Vector2(p).rotate(angle) for p in points]

为了计算最终得分，需要通过三角形的中心对点进行缩放和平移：

triangle_points = [(vCenter + p*scale) for p in rotated_point]

查看示例：

import pygame
import math

def rotate_triangle(center, scale, mouse_pos):

    vMouse  = pygame.math.Vector2(mouse_pos)
    vCenter = pygame.math.Vector2(center)
    angle   = pygame.math.Vector2().angle_to(vMouse - vCenter)

    points = [(-0.5, -0.866), (-0.5, 0.866), (2.0, 0.0)]
    rotated_point = [pygame.math.Vector2(p).rotate(angle) for p in points]

    triangle_points = [(vCenter + p*scale) for p in rotated_point]
    return triangle_points

disp=pygame.display.set_mode((200,200))

run = True
while run:

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            run = False

    mouse_position = pygame.mouse.get_pos()
    points = rotate_triangle((100, 100), 10, mouse_position)

    pygame.Surface.fill(disp, (255,255,255))
    pygame.draw.polygon(disp, (0,0,0), points)
    pygame.display.update()

---

一个不使用pygame.math.Vector2的算法版本如下：

def rotate_triangle(center, scale, mouse_pos):

    dx = mouse_pos[0] - center[0]
    dy = mouse_pos[1] - center[1]
    len = math.sqrt(dx*dx + dy*dy)
    dx, dy = (dx*scale/len, dy*scale/len) if len > 0 else (1, 0)

    pts = [(-0.5, -0.866), (-0.5, 0.866), (2.0, 0.0)]
    pts = [(center[0] + p[0]*dx + p[1]*dy, center[1] + p[0]*dy - p[1]*dx) for p in pts]
    return pts

注意，这个版本可能会更快。与.angle_to()和math.sin分别使用的math.atan2和math.cos相比，它需要进行一个math.sqrt操作。

结果坐标是相同的。

- Rabbid76