2016-11-23 210 views
0

我正在研究一個程序,我需要兩個不同的圖形進行動畫。我無法理解如何使用我正在使用的結構來做到這一點。我會在下面粘貼我的代碼,以便您可以嘗試。我儘可能地將其剝離,同時仍然保留核心功能,因此希望不會太難理解。在目前的狀態下,動畫線沒有做任何事情,所以請讓我知道我出錯的地方。Tkinter/Matplotlib動畫幫助

from Tkinter import *    #Used for GUI elements 
import time       #Used for timing elements 
import matplotlib     #Used for graphing 
matplotlib.use("TkAgg") 
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2TkAgg 
from matplotlib.figure import Figure 
import matplotlib.animation as animation 
import numpy as np     #Used for arrays to find min/max of float array 
import random      #Only used for fake data input 


class tkgui: 
    def __init__(self, parent): 
     #--------------The following are variables that need to be accessed by other functions---------------------- 
     #Raw input values 
     self.x = 209500 
     self.y = 0 
     self.timeElapsed = 0 

     #State values 
     self.curFrame = 1 

     #List Values 
     self.timeList = np.array([]) 
     self.yList = np.array([]) 
     self.xList = np.array([]) 

     self.lastX = 0 
     #---------------------------------------------------------------------------------------------------------- 

     #Make Tkinter fullscreen 
     w, h = 320,240 #int(str(root.winfo_screenwidth())), int(str(root.winfo_screenheight())) #320, 240 is the RPiTFT 

     #The base layer of the GUI 
     topLevelContainer = Frame(parent) 
     topLevelContainer.pack() 

     #The two 'screens' to switch between. They contain everything on the GUI 
     self.buttonsFrame = Frame(topLevelContainer) 
     self.graphFrame = Frame(topLevelContainer) 

     #Stack the frames so that they are switchable 
     for frame in self.buttonsFrame, self.graphFrame: 
      frame.grid(row=0, column=0, sticky='news', padx=5, pady=(10, 10)) 

     buttonsFrameButtons = Frame(self.buttonsFrame) 
     buttonsFrameButtons.pack(side=LEFT, padx=(0, 50)) 

     #X button 
     self.xButton = Button(buttonsFrameButtons, command=self.xButtonClick) 
     self.xButton.configure(text="X", background="#C8C8C8", width=6, padx=35, pady=35) 
     self.xButton.pack(side=TOP, pady=10) 

     #Y button 
     self.yButton = Button(buttonsFrameButtons, command=self.yButtonClick) 
     self.yButton.configure(text="Y", background="#C8C8C8", width=6, padx=35, pady=35) 
     self.yButton.pack(side=TOP, pady=10) 

     #Bar graph 
     buttonsFrameBar = Frame(self.buttonsFrame) 
     buttonsFrameBar.pack(side=LEFT) 

     self.fBar = Figure(figsize=(2, 4), dpi=50) 
     aBar = self.fBar.add_subplot(111) 
     self.xBar = aBar.bar([0, 1], [0, 0], width=1) 

     lAxes = self.fBar.gca() 
     lAxes.axes.get_xaxis().set_ticklabels([]) 

     aBar.set_ylim([-30000, 30000]) 
     self.fBar.tight_layout() 

     self.buttonsFrame.tkraise()   

     #Setup the matplotlib graph 
     self.fGraph = Figure(figsize=(5, 3), dpi=50) 
     #Create the Y axis 
     aGraph = self.fGraph.add_subplot(111) 
     aGraph.set_xlabel("Time (s)") 
     aGraph.set_ylabel("Y") 
     self.yLine, = aGraph.plot([],[], "r-") 

     #Create the X axis 
     a2Graph = aGraph.twinx() 
     self.xLine, = a2Graph.plot([], []) 
     a2Graph.set_ylabel("X") 

     #Final matplotlib/Tkinter setup 
     self.canvasGraph = FigureCanvasTkAgg(self.fGraph, master=self.graphFrame) 
     self.canvasGraph.show() 
     self.canvasGraph.get_tk_widget().pack(side=LEFT, fill=BOTH, expand=1) 

     self.canvasBar = FigureCanvasTkAgg(self.fBar, master=buttonsFrameBar) 
     self.canvasBar.show() 
     self.canvasBar.get_tk_widget().pack(side=BOTTOM, fill=BOTH, expand=1) 

     #Resize the plot to fit all of the labels in 
     self.fGraph.subplots_adjust(bottom=0.13, left=0.15, right=0.87)  

    def refreshGraph(self, frameno):  #Redraw the graph with the updated arrays and resize it accordingly 
     #Update data 
     self.yLine.set_data(self.timeList, self.yList) 
     self.xLine.set_data(self.timeList, self.xList) 

     #Update y axis 
     ax = self.canvasGraph.figure.axes[0] 
     ax.set_xlim(self.timeList.min(), self.timeList.max()) 
     ax.set_ylim(self.yList.min(), self.yList.max()) 

     #Update x axis 
     ax2 = self.canvasGraph.figure.axes[1] 
     ax2.set_xlim(self.timeList.min(), self.timeList.max()) 
     ax2.set_ylim(self.xList.min(), self.xList.max()) 

     #Redraw 
     self.canvasGraph.draw() 


    def refreshBar(self, frameno): 
     curX = self.x 
     dif = curX - self.lastX 
     i = [dif] 
     for rect, h in zip(self.xBar, i): 
      rect.set_height(h) 
      if dif > 0: 
       rect.set_color('b') 
      else: 
       rect.set_color('r') 

     self.canvasBar.draw() 
     self.lastX=curX 

    def switchFrame(self):  #Switch the current screen. Either x/y buttons or graph 
     if self.curFrame: 
      self.graphFrame.tkraise() 
      self.curFrame = 0 
     else: 
      self.buttonsFrame.tkraise() 
      self.curFrame = 1 

    def xButtonClick(self): 
     self.switchFrame() 

    def yButtonClick(self): 
     self.close() 

    def close(e):    #Exit the program 
     sys.exit() 

#Initialisation of global variables 
lastTime = 0  #Used for the 'last time' iterated 
yState = 0  

def updateNumbers():  #Used to generate fake input variables. Will be replaced by ADC values 
    global lastTime 
    global yState 

    curTime = time.time()           #Update the time each time the function is called 
    if curTime - lastTime > 0.5:         #Only update numbers every 0.5 seconds 
     gui.x = random.randrange(200000, 230000)     #Generates x 
     if yState: 
      gui.y = gui.y - 20        #Decrease y 
      if gui.y < 1: 
       yState = 0          #Until it gets to a minimum of 0 
     else: 
      gui.y = gui.y + 20        #Increase y 
      if gui.y > 1300: 
       yState = 1          #Until it reaches a maximum of 1300 
     gui.yList = np.append(gui.yList, gui.y)   #Add the new y values to the array 
     gui.xList = np.append(gui.xList, gui.x/10000.0)   #Add the new x values to the array 
     lastTime = time.time()          #Record the last time iterated for timing purposes 
     gui.timeElapsed += 0.5          
     gui.timeList = np.append(gui.timeList, gui.timeElapsed)  #Add the latest time to the array 
##  gui.refreshGraph()           #Call the function that will redraw the graph with the new figures 

if __name__ == "__main__": 
    root = Tk()   #Root Tkinter setup 
    gui = tkgui(root) #Setup the gui class 

    updateNumbers() 
    aniGraph = animation.FuncAnimation(gui.fGraph,gui.refreshGraph,interval=500,frames=100,repeat=True) 
    aniBar = animation.FuncAnimation(gui.fBar,gui.refreshBar,interval=500,frames=100,repeat=True) 

    while(1):     #Main loop 
     updateNumbers()   #Update fake values 

     root.update()   #Update the gui loop 

    root.mainloop()    #Tkinter main loop 

爲了清楚起見,我只問了如何讓動畫爲這段代碼工作。

+0

使用after,而(1):''然後root.mainloop( )'永遠不會被使用。 'updateNumber'結束時,你可以使用'root.after(500,updateNumber)',它會在500ms(0.5s)後再次運行'updateNumber',所以你不需要'while(1)'循環。 – furas

+0

在開始我看到酒吧動畫和點擊X按鈕後,我看到線動畫 - 它的作品。我不知道是什麼問題。 – furas

+0

這個版本的代碼沒有顯示,但我需要運行其他原因的恆定循環。我以爲while(1)裏面的root.update()修復了這個問題呢?你有沒有改變任何東西,因爲當我運行相同的代碼時,我看不到任何動畫。 –

回答

1

你的代碼適用於我 - 我看到所有的動畫 - 但如果你運行它沒有While(1):(或更pythonic While True:),那麼你可以使用root.after(milliseconds, function_name)。你可以使用它來代替FuncAnimation

它可以讓你控制功能 - 啓動/停止它。

if self.run_bar: 
    root.after(100, self.refreshBar) 

您可以使用啓動(或重啓)

self.run_bar = True 
self.refreshBar() 

,你可以阻止它

self.run_bar = False 

查看所有# <--代碼。

from Tkinter import *    #Used for GUI elements 
import time       #Used for timing elements 
import matplotlib     #Used for graphing 
matplotlib.use("TkAgg") 
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2TkAgg 
from matplotlib.figure import Figure 
import matplotlib.animation as animation 
import numpy as np     #Used for arrays to find min/max of float array 
import random      #Only used for fake data input 


class TkGUI: # <-- CamelCase names for classes 
      # <-- empty line for readabelity 
    def __init__(self, parent): 
     #--- The following are variables that need to be accessed by other functions---------------------- 
     #Raw input values 
     self.x = 209500 
     self.y = 0 
     self.timeElapsed = 0 

     #State values 
     self.curFrame = 1 

     #List Values 
     self.timeList = np.array([]) 
     self.yList = np.array([]) 
     self.xList = np.array([]) 

     self.lastX = 0 
     #----------------------------------------------------------- 

     #Make Tkinter fullscreen 
     w, h = 320,240 #int(str(root.winfo_screenwidth())), int(str(root.winfo_screenheight())) #320, 240 is the RPiTFT 

     #The base layer of the GUI 
     topLevelContainer = Frame(parent) 
     topLevelContainer.pack() 

     #The two 'screens' to switch between. They contain everything on the GUI 
     self.buttonsFrame = Frame(topLevelContainer) 
     self.graphFrame = Frame(topLevelContainer) 

     #Stack the frames so that they are switchable 
     for frame in self.buttonsFrame, self.graphFrame: 
      frame.grid(row=0, column=0, sticky='news', padx=5, pady=(10, 10)) 

     buttonsFrameButtons = Frame(self.buttonsFrame) 
     buttonsFrameButtons.pack(side=LEFT, padx=(0, 50)) 

     #X button 
     self.xButton = Button(buttonsFrameButtons, command=self.xButtonClick) 
     self.xButton.configure(text="X", background="#C8C8C8", width=6, padx=35, pady=35) 
     self.xButton.pack(side=TOP, pady=10) 

     #Y button 
     self.yButton = Button(buttonsFrameButtons, command=self.yButtonClick) 
     self.yButton.configure(text="Y", background="#C8C8C8", width=6, padx=35, pady=35) 
     self.yButton.pack(side=TOP, pady=10) 

     #Bar graph 
     buttonsFrameBar = Frame(self.buttonsFrame) 
     buttonsFrameBar.pack(side=LEFT) 

     self.fBar = Figure(figsize=(2, 4), dpi=50) 
     aBar = self.fBar.add_subplot(111) 
     self.xBar = aBar.bar([0, 1], [0, 0], width=1) 

     lAxes = self.fBar.gca() 
     lAxes.axes.get_xaxis().set_ticklabels([]) 

     aBar.set_ylim([-30000, 30000]) 
     self.fBar.tight_layout() 

     self.buttonsFrame.tkraise()   

     #Setup the matplotlib graph 
     self.fGraph = Figure(figsize=(5, 3), dpi=50) 
     #Create the Y axis 
     aGraph = self.fGraph.add_subplot(111) 
     aGraph.set_xlabel("Time (s)") 
     aGraph.set_ylabel("Y") 
     self.yLine, = aGraph.plot([],[], "r-") 

     #Create the X axis 
     a2Graph = aGraph.twinx() 
     self.xLine, = a2Graph.plot([], []) 
     a2Graph.set_ylabel("X") 

     #Final matplotlib/Tkinter setup 
     self.canvasGraph = FigureCanvasTkAgg(self.fGraph, master=self.graphFrame) 
     self.canvasGraph.show() 
     self.canvasGraph.get_tk_widget().pack(side=LEFT, fill=BOTH, expand=1) 

     self.canvasBar = FigureCanvasTkAgg(self.fBar, master=buttonsFrameBar) 
     self.canvasBar.show() 
     self.canvasBar.get_tk_widget().pack(side=BOTTOM, fill=BOTH, expand=1) 

     #Resize the plot to fit all of the labels in 
     self.fGraph.subplots_adjust(bottom=0.13, left=0.15, right=0.87)  

    def refreshGraph(self): # <-- without argument 
     '''Redraw the graph with the updated arrays and resize it accordingly''' # <-- docstring used by documentation generator and IDE as help 

     #Update data 
     self.yLine.set_data(self.timeList, self.yList) 
     self.xLine.set_data(self.timeList, self.xList) 

     #Update y axis 
     ax = self.canvasGraph.figure.axes[0] 
     ax.set_xlim(self.timeList.min(), self.timeList.max()) 
     ax.set_ylim(self.yList.min(), self.yList.max()) 

     #Update x axis 
     ax2 = self.canvasGraph.figure.axes[1] 
     ax2.set_xlim(self.timeList.min(), self.timeList.max()) 
     ax2.set_ylim(self.xList.min(), self.xList.max()) 

     #Redraw 
     self.canvasGraph.draw() 

     # run again after 100ms (0.1s) 
     root.after(100, self.refreshGraph) # <-- run again like in loop 

    def refreshBar(self): # <-- without argument 
     curX = self.x 
     dif = curX - self.lastX 
     i = [dif] 
     for rect, h in zip(self.xBar, i): 
      rect.set_height(h) 
      if dif > 0: 
       rect.set_color('b') 
      else: 
       rect.set_color('r') 

     self.canvasBar.draw() 
     self.lastX=curX 
     # run again after 100ms (0.1s) 
     root.after(100, self.refreshBar) # <-- run again like in loop 

    def switchFrame(self):  #Switch the current screen. Either x/y buttons or graph 
     if self.curFrame: 
      self.graphFrame.tkraise() 
      self.curFrame = 0 
     else: 
      self.buttonsFrame.tkraise() 
      self.curFrame = 1 

    def xButtonClick(self): 
     self.switchFrame() 

    def yButtonClick(self): 
     self.close() 

    def close(e): # Exit the program 
     sys.exit() 

#Initialisation of global variables 
lastTime = 0  #Used for the 'last time' iterated 
yState = 0  

def updateNumbers():  #Used to generate fake input variables. Will be replaced by ADC values 
    global lastTime 
    global yState 

    curTime = time.time()           #Update the time each time the function is called 
    if curTime - lastTime > 0.5:         #Only update numbers every 0.5 seconds 
     gui.x = random.randrange(200000, 230000)     #Generates x 
     if yState: 
      gui.y = gui.y - 20        #Decrease y 
      if gui.y < 1: 
       yState = 0          #Until it gets to a minimum of 0 
     else: 
      gui.y = gui.y + 20        #Increase y 
      if gui.y > 1300: 
       yState = 1          #Until it reaches a maximum of 1300 
     gui.yList = np.append(gui.yList, gui.y)   #Add the new y values to the array 
     gui.xList = np.append(gui.xList, gui.x/10000.0)   #Add the new x values to the array 
     lastTime = time.time()          #Record the last time iterated for timing purposes 
     gui.timeElapsed += 0.5          
     gui.timeList = np.append(gui.timeList, gui.timeElapsed)  #Add the latest time to the array 

    # run again after 100ms (0.1s) 
    root.after(100, updateNumbers) # <-- run again like in loop  

if __name__ == "__main__": 
    root = Tk() 
    gui = TkGUI(root) 

    # <-- vvv - without While and without FuncAnimation - vvv 

    updateNumbers()  # run first time 

    gui.refreshBar() # run first time 
    gui.refreshGraph() # run first time 

    # <-- ^^^ - without While and without FuncAnimation - ^^^ 

    root.mainloop()  # Tkinter main loop 

編輯:當然你可以保持FuncAnimation沒有after,如果你使用`只有在updateNumbers

from Tkinter import *    #Used for GUI elements 
import time       #Used for timing elements 
import matplotlib     #Used for graphing 
matplotlib.use("TkAgg") 
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2TkAgg 
from matplotlib.figure import Figure 
import matplotlib.animation as animation 
import numpy as np     #Used for arrays to find min/max of float array 
import random      #Only used for fake data input 


class TkGUI: # <-- CamelCase names for classes 
      # <-- empty line for readabelity 
    def __init__(self, parent): 
     #--- The following are variables that need to be accessed by other functions---------------------- 
     #Raw input values 
     self.x = 209500 
     self.y = 0 
     self.timeElapsed = 0 

     #State values 
     self.curFrame = 1 

     #List Values 
     self.timeList = np.array([]) 
     self.yList = np.array([]) 
     self.xList = np.array([]) 

     self.lastX = 0 
     #----------------------------------------------------------- 

     #Make Tkinter fullscreen 
     w, h = 320,240 #int(str(root.winfo_screenwidth())), int(str(root.winfo_screenheight())) #320, 240 is the RPiTFT 

     #The base layer of the GUI 
     topLevelContainer = Frame(parent) 
     topLevelContainer.pack() 

     #The two 'screens' to switch between. They contain everything on the GUI 
     self.buttonsFrame = Frame(topLevelContainer) 
     self.graphFrame = Frame(topLevelContainer) 

     #Stack the frames so that they are switchable 
     for frame in self.buttonsFrame, self.graphFrame: 
      frame.grid(row=0, column=0, sticky='news', padx=5, pady=(10, 10)) 

     buttonsFrameButtons = Frame(self.buttonsFrame) 
     buttonsFrameButtons.pack(side=LEFT, padx=(0, 50)) 

     #X button 
     self.xButton = Button(buttonsFrameButtons, command=self.xButtonClick) 
     self.xButton.configure(text="X", background="#C8C8C8", width=6, padx=35, pady=35) 
     self.xButton.pack(side=TOP, pady=10) 

     #Y button 
     self.yButton = Button(buttonsFrameButtons, command=self.yButtonClick) 
     self.yButton.configure(text="Y", background="#C8C8C8", width=6, padx=35, pady=35) 
     self.yButton.pack(side=TOP, pady=10) 

     #Bar graph 
     buttonsFrameBar = Frame(self.buttonsFrame) 
     buttonsFrameBar.pack(side=LEFT) 

     self.fBar = Figure(figsize=(2, 4), dpi=50) 
     aBar = self.fBar.add_subplot(111) 
     self.xBar = aBar.bar([0, 1], [0, 0], width=1) 

     lAxes = self.fBar.gca() 
     lAxes.axes.get_xaxis().set_ticklabels([]) 

     aBar.set_ylim([-30000, 30000]) 
     self.fBar.tight_layout() 

     self.buttonsFrame.tkraise()   

     #Setup the matplotlib graph 
     self.fGraph = Figure(figsize=(5, 3), dpi=50) 
     #Create the Y axis 
     aGraph = self.fGraph.add_subplot(111) 
     aGraph.set_xlabel("Time (s)") 
     aGraph.set_ylabel("Y") 
     self.yLine, = aGraph.plot([],[], "r-") 

     #Create the X axis 
     a2Graph = aGraph.twinx() 
     self.xLine, = a2Graph.plot([], []) 
     a2Graph.set_ylabel("X") 

     #Final matplotlib/Tkinter setup 
     self.canvasGraph = FigureCanvasTkAgg(self.fGraph, master=self.graphFrame) 
     self.canvasGraph.show() 
     self.canvasGraph.get_tk_widget().pack(side=LEFT, fill=BOTH, expand=1) 

     self.canvasBar = FigureCanvasTkAgg(self.fBar, master=buttonsFrameBar) 
     self.canvasBar.show() 
     self.canvasBar.get_tk_widget().pack(side=BOTTOM, fill=BOTH, expand=1) 

     #Resize the plot to fit all of the labels in 
     self.fGraph.subplots_adjust(bottom=0.13, left=0.15, right=0.87)  

    def refreshGraph(self, i): 
     '''Redraw the graph with the updated arrays and resize it accordingly''' # <-- docstring used by documentation generator and IDE as help 

     #Update data 
     self.yLine.set_data(self.timeList, self.yList) 
     self.xLine.set_data(self.timeList, self.xList) 

     #Update y axis 
     ax = self.canvasGraph.figure.axes[0] 
     ax.set_xlim(self.timeList.min(), self.timeList.max()) 
     ax.set_ylim(self.yList.min(), self.yList.max()) 

     #Update x axis 
     ax2 = self.canvasGraph.figure.axes[1] 
     ax2.set_xlim(self.timeList.min(), self.timeList.max()) 
     ax2.set_ylim(self.xList.min(), self.xList.max()) 

     #Redraw 
     self.canvasGraph.draw() 

    def refreshBar(self, i): 
     curX = self.x 
     dif = curX - self.lastX 
     i = [dif] 
     for rect, h in zip(self.xBar, i): 
      rect.set_height(h) 
      if dif > 0: 
       rect.set_color('b') 
      else: 
       rect.set_color('r') 

     self.canvasBar.draw() 
     self.lastX=curX 

    def switchFrame(self): 
     '''Switch the current screen. Either x/y buttons or graph''' 

     if self.curFrame: 
      self.graphFrame.tkraise() 
      self.curFrame = 0 
     else: 
      self.buttonsFrame.tkraise() 
      self.curFrame = 1 

    def xButtonClick(self): 
     self.switchFrame() 

    def yButtonClick(self): 
     self.close() 

    def close(e): # Exit the program 
     sys.exit() 

#Initialisation of global variables 
lastTime = 0  #Used for the 'last time' iterated 
yState = 0  

def updateNumbers():  #Used to generate fake input variables. Will be replaced by ADC values 
    global lastTime 
    global yState 

    curTime = time.time()           #Update the time each time the function is called 
    if curTime - lastTime > 0.5:         #Only update numbers every 0.5 seconds 
     gui.x = random.randrange(200000, 230000)     #Generates x 
     if yState: 
      gui.y = gui.y - 20        #Decrease y 
      if gui.y < 1: 
       yState = 0          #Until it gets to a minimum of 0 
     else: 
      gui.y = gui.y + 20        #Increase y 
      if gui.y > 1300: 
       yState = 1          #Until it reaches a maximum of 1300 
     gui.yList = np.append(gui.yList, gui.y)   #Add the new y values to the array 
     gui.xList = np.append(gui.xList, gui.x/10000.0)   #Add the new x values to the array 
     lastTime = time.time()          #Record the last time iterated for timing purposes 
     gui.timeElapsed += 0.5          
     gui.timeList = np.append(gui.timeList, gui.timeElapsed)  #Add the latest time to the array 

    # run again after 100ms (0.1s) 
    root.after(100, updateNumbers) # <-- run again like in loop  

if __name__ == "__main__": 
    root = Tk() 
    gui = TkGUI(root) 

    aniGraph = animation.FuncAnimation(gui.fGraph,gui.refreshGraph,interval=500,frames=100,repeat=True) 
    aniBar = animation.FuncAnimation(gui.fBar,gui.refreshBar,interval=500,frames=100,repeat=True) 

    # <-- vvv - without While - vvv 

    updateNumbers()  # run first time 

    # <-- ^^^ - without While - ^^^ 

    root.mainloop()  # Tkinter main loop 
+0

謝謝,這將工作。然而我想知道是否使用matplotlib動畫的東西比重新繪製更加優化。我這樣說是因爲在我的主代碼中,當我開始獲取大數據集和長圖時,或者如果我更快地更新它,事情會開始運行緩慢。 –

+0

我從不檢查兩種方法的速度,但我沒有更改'refreshBar'和'refreshGraph'中的任何內容,所以兩種方法都應該以相同的速度工作。也許如果你檢查FuncAnimation源代碼,那麼你也許會看到'after()':)'matplotlib/FuncAnimation'與'tkinter'一起使用必須使用'tkinter'函數。 – furas

+0

順便說一句:你可以在''after'後面保留'FuncAnimation',並且''在'updateNumbers'後面使用'after' - 它也可以。 – furas