Анимация Python, линия, соединяющая две движущиеся точки и остановка на пересечении стационарной линии - PullRequest
Новая
       21

Анимация Python, линия, соединяющая две движущиеся точки и остановка на пересечении стационарной линии

0 голосов
/ 24 января 2019

Я делаю программу для имитации ретроградного движения Марса, видимого с Земли. Так что это вид сверху Земли и Марса, вращающихся вокруг Солнца. Существует также линия, идущая от Земли к Марсу. Однако мне нужно, чтобы он пересек точку Марса и продолжал двигаться до пересечения линии x = 15

.

другой пользователь прислал мне этот код с

Мой предыдущий код 

import math
import matplotlib.pyplot as plt
import matplotlib.animation as animation
def _update_plot(i, fig, scat, l):
    scat.set_offsets(([math.cos(math.radians(i))*5, math.sin(math.radians(i))*5], [math.cos(math.radians(i/2))*10, math.sin(math.radians(i/2))*10], [0, 0]))
        l.set_data(([math.cos(math.radians(i))*5,math.cos(math.radians(i/2))*10],[math.sin(math.radians(i))*5,math.sin(math.radians(i/2))*10]))
    return [scat,l]

fig = plt.figure()

x = [0]
y = [0]

ax = fig.add_subplot(111)
ax.set_aspect('equal')
ax.grid(True, linestyle = '-', color = '0.10')
ax.set_xlim([-11, 11])
ax.set_ylim([-11, 11])

l, = plt.plot([],[], 'r--', zorder=1)
scat = plt.scatter(x, y, c = x, zorder=2)
scat.set_alpha(0.8)

anim = animation.FuncAnimation(fig, _update_plot, fargs = (fig, scat, l),
                           frames = 720, interval = 10)

plt.show()

код другого пользователя 

def _update_plot(i, fig, scat, l):
    angle = math.radians(i)
    sun_pos = np.array([0., 0.])
    earth_pos = np.array([math.cos(angle)*5, math.sin(angle)*5])
    mars_pos = np.array([math.cos(angle / 2.) * 10, math.sin(angle / 2.) * 10])

    # compute the unit vector that points from Earth to Mars
    direction = mars_pos - earth_pos
    direction /= np.sqrt(np.dot(direction, direction))

    # find out where the line would intersect the x = 15 axis
    start_from = earth_pos
    alpha = (15 - start_from[0]) / direction[0]
    # if alpha comes out to be negative we picked the "wrong" planet
    if alpha < 0:
            start_from = mars_pos
            direction = -direction
            alpha = (15 - start_from[0]) / direction[0]
    y_line = start_from[1] + alpha * direction[1]

    # draw the planets
    scat.set_offsets((earth_pos, mars_pos, sun_pos))
    # draw the line
    l.set_data(([start_from[0], 15], [start_from[1], y_line]))
    return [scat,l]

Таким образом, цель в том, чтобы была линия, идущая от «Земли» (самый внутренний движущийся объект) к линии х = 15, а также другой движущийся объект между ними.

Я также предполагаю, что линия должна появиться только тогда, когда линия достигнет х = 15, поэтому только в течение первой и последней четверти периода орбиты

...