Pygame Vector2.as_polar() и Vector2.from_polar() методы

Question

Может кто-нибудь, пожалуйста, покажите мне пример использования Vector2.as_polar() чтобы получить расстояние и азимутальный угол, а затем подключить эту информацию обратно в Vector2.from_polar() получить декартовы координаты?

Давайте предположим, что мы хотим расстояние и азимутальный угол между (0, 0) а также (15, 10), а затем вставьте эту информацию в Vector2.from_polar() чтобы генерировать (15, 10) снова.

Я бы точно объяснил, что я делаю, но мне и всем, возможно, будет проще увидеть рабочий пример.

Answer 1

Вы можете определить вектор как радиус, а угол - как полярные координаты.

Давайте использовать точечный вектор в центре экрана как полюс нашей полярной системы координат.

pole = Vector2(screen_rect.center)

Установить координаты вектора с помощью from_polar Метод, сначала определите вектор, а затем передайте кортеж, который состоит из радиуса и угла:

vec = Vector2()
# This updates the cartesian coordinates of vec.
vec.from_polar((90, 60))  # 90 pixels long, rotated 60 degrees.

Тогда используйте pygame.draw.line нарисовать это (добавить vec в pole чтобы получить целевые координаты):

pg.draw.line(screen, (200, 90, 20), pole, pole+vec, 3)

---

В качестве демонстрации для as_polar, рассчитать вектор до положения мыши и распаковать полярные координаты в переменные r а также phi который мы визуализируем позже, чтобы увидеть текущий радиус и угол этого вектора.

r, phi = (mouse_pos-pole).as_polar()

Обратите внимание, что ось Y в Pygame перевернута, поэтому отрицательные углы направлены вверх, а положительные углы вниз.

import sys
import pygame as pg
from pygame.math import Vector2


def main():
    screen = pg.display.set_mode((640, 480))
    screen_rect = screen.get_rect()
    font = pg.font.Font(None, 30)
    color = pg.Color(150, 250, 100)
    clock = pg.time.Clock()
    pole = Vector2(screen_rect.center)
    # A zero vector.
    vec = Vector2()
    # To set the coordinates of `vec` we can pass the
    # polar coordinates (radius, angle) to `from_polar`.
    vec.from_polar((90, 60))
    print(vec)  # Now take a look at the updated cartesian coordinates.
    done = False

    while not done:
        for event in pg.event.get():
            if event.type == pg.QUIT:
                done = True

        mouse_pos = pg.mouse.get_pos()
        # `as_polar` gives us the radius and angle of the vector that
        # points to the mouse. Unpack it into the `r` and `phi` variables.
        r, phi = (mouse_pos-pole).as_polar()

        screen.fill((30, 30, 30))
        # Draw the line to the mouse pos.
        pg.draw.line(screen, color, pole, mouse_pos, 3)
        # Add the vec to the pole to get the target coordinates.
        pg.draw.line(screen, (200, 90, 20), pole, pole+vec, 3)

        # Render the radius and angle.
        txt = font.render('r: {:.1f}'.format(r), True, color)
        screen.blit(txt, (30, 30))
        txt = font.render('phi: {:.1f}'.format(phi), True, color)
        screen.blit(txt, (30, 50))

        pg.display.flip()
        clock.tick(30)


if __name__ == '__main__':
    pg.init()
    main()
    pg.quit()
    sys.exit()

---

Вот еще один пример со спрайтом, который следует за мышью (нажмите "w" или "s" для ускорения). Я передаю self.speed как радиус и угол, as_polar возвращает в качестве аргумента from_polar установить velГород игрока спрайт.

import sys
import pygame as pg


class Player(pg.sprite.Sprite):
    def __init__(self, pos):
        super().__init__()
        self.image = pg.Surface((50, 30), pg.SRCALPHA)
        color = pg.Color('dodgerblue1')
        pg.draw.polygon(self.image, color, ((1, 1), (49, 15), (1, 29)))
        self.orig_img = self.image
        self.rect = self.image.get_rect(center=pos)
        self.pos = pg.math.Vector2(pos)
        self.vel = pg.math.Vector2(0, 0)
        self.speed = 0

    def update(self):
        self.rotate()
        self.pos += self.vel
        self.rect.center = self.pos

    def rotate(self):
        _, angle = (pg.mouse.get_pos()-self.pos).as_polar()
        self.vel.from_polar((self.speed, angle))
        self.image = pg.transform.rotozoom(self.orig_img, -angle, 1)
        self.rect = self.image.get_rect(center=self.rect.center)


def main():
    screen = pg.display.set_mode((640, 480))
    clock = pg.time.Clock()
    all_sprites = pg.sprite.Group()
    player = Player((300, 200))
    all_sprites.add(player)

    done = False
    while not done:
        for event in pg.event.get():
            if event.type == pg.QUIT:
                done = True

        keys = pg.key.get_pressed()
        if keys[pg.K_w]:
            player.speed += .2
        elif keys[pg.K_s]:
            player.speed -= .2

        all_sprites.update()
        screen.fill((30, 30, 30))
        all_sprites.draw(screen)

        pg.display.flip()
        clock.tick(30)


if __name__ == '__main__':
    pg.init()
    main()
    pg.quit()
    sys.exit()