2017-04-13 54 views
1

自分のイベントループで実行しているアニメーションを一時停止します。matplotlibカスタムアニメーションループを一時停止します

import numpy as np 
import matplotlib.pyplot as plt 
from matplotlib.widgets import Button 
from mpl_toolkits.mplot3d import proj3d 

import time 

def main(): 
    global paused 
    paused = False 

    fig = plt.figure() 
    ax = fig.add_subplot(111, projection='3d') 

    ax.set_ylim(-100, 100) 
    ax.set_xlim(-10, 10) 
    ax.set_zlim(-100, 100) 

    plt.ion() 
    plt.show() 

    def pause_anim(event): 
     global paused 
     paused = not paused 

    pause_ax = fig.add_axes((0.7, 0.03, 0.1, 0.04)) 
    pause_button = Button(pause_ax, 'pause', hovercolor='0.975') 
    pause_button.on_clicked(pause_anim) 

    x = np.arange(-50, 51) 

    line = ax.plot([], [], [], c="r")[0] 

    y_range = list(np.arange(1, 60, 3)) 
    y_len = len(y_range) 
    idx = 0 

    while True: 

     if not paused: 
      idx += 1 
      if idx >= y_len: 
       break 

      y = y_range[idx] 
      z = - x**2 + y - 100 

      line.set_data(x, 0) 
      line.set_3d_properties(z) 

      plt.draw() 
      plt.pause(0.2) 

     else: 
      time.sleep(1) # this stops button events from happening 
      #input("Shoop?") # prompting for input works 
      # I've also tried putting a mutex here 

if __name__ == '__main__': 
    main() 

私はコードで言及したように、私はtime.sleepLockを試してみたが、これらは私が一時停止したら一時停止の解除から私を停止します。ここでは、コードの簡易版です。アニメーションを再開する機能を解除せずにループを一時停止するにはどうすればよいですか?

答えて

1

elseの部分のtime.sleep(1)は、ifの部分と同じように、plt.pauseと置き換えることができます。

 .... 
     plt.draw() 
     plt.pause(0.2) 
    else: 
     plt.pause(0.2) 

他のオプションは、メソッド.event_source.stop().event_source.start()を有し、FuncAnimationのように、matplotlibのアニメーションを使用することはもちろんです。 (この質問に対する回答の2つに示されているように:stop/start/pause in python matplotlib animation

関連する問題