Слайдер в прямом эфире меняется Wave Packet анимация Matplotlib - PullRequest
Новая
       21

Слайдер в прямом эфире меняется Wave Packet анимация Matplotlib

0 голосов
/ 23 апреля 2020

Моя анимация медленная из-за функции обновления, которая постоянно вычисляет матрицу, которая изменяется только при перемещении ползунка: 

def evolve(psi0, height, width):
    global psi
    potential = potentialBarrier(height, width)
    tM = timeMatrix(hamiltonian(potential))
    psi = tM.dot(psi0)
    presenceProbability = abs(psi) ** 2
    norm = sum(presenceProbability)
    presenceProbability /= norm
    psi /= norm ** 0.5
    return presenceProbability


def update(frame):
    line.set_ydata(evolve(psi, potentialSlider.val*10, widthSlider.val))
    line2.set_ydata(potentialBarrier(potentialSlider.val, widthSlider.val))
    return

wavePacketAnimation = animation.FuncAnimation(fig, update, interval=5, blit=False)

Как я могу заставить функцию обновления вычислять новый результат моей матрицы только , когда я перемещаю значение ползунков и использую предыдущий результат, когда я не перемещаю ползунки?

Я пытался определить глобальную матрицу, но это не сработало. Я получил это до сих пор 

sliderHasChanged = True
tM = timeMatrix(hamiltonian(potentialBarrierInit))


def evolve(psi0, height, width):
    global tM
    global sliderHasChanged
    if sliderHasChanged:
        potential = potentialBarrier(height * 10, width)
        tM = timeMatrix(hamiltonian(potential))
        sliderHasChanged = False
    global psi 
    psi = tM.dot(psi0)
    presenceProbability = abs(psi) ** 2
    norm = sum(presenceProbability)
    presenceProbability /= norm
    psi /= norm ** 0.5
    return presenceProbability


potentialSlider_ax = fig.add_axes([0.25, 0.15, 0.65, 0.03])
potentialSlider = Slider(potentialSlider_ax, '$V_0$', 0, 1, valinit=0.1)
potentialSlider.label.set_size(15)

widthSlider_ax = fig.add_axes([0.25, 0.1, 0.65, 0.03])
widthSlider = Slider(widthSlider_ax, '$a$', 0, 20, valinit=10.0)
widthSlider.label.set_size(15)


def sliderChange():
    global sliderHasChanged
    sliderHasChanged = True
    return


potentialSlider.on_changed(sliderChange())
widthSlider.on_changed(sliderChange())


def update(frame):
    line.set_ydata(evolve(psi, potentialSlider.val, widthSlider.val))
    line2.set_ydata(potentialBarrier(potentialSlider.val, widthSlider.val))
    return


wavePacketAnimation = animation.FuncAnimation(fig, update, interval=5, blit=False)
wavePacketAnimation.running = True

Анимация хорошо работает с исходным значением матрицы, но изменение в ползунке не влияет на анимацию

...