Использование FigureCanvasTkAgg на двух страницах TKinter с python

Я пытаюсь получить две «страницы» TKinter, каждая из которых имеет другой сюжет, обновленный с помощью функции анимации matplotlib.

Проблема в том, что только одна из страниц показывает правильно сюжет.

Вот код:

import matplotlib
matplotlib.use("TkAgg")

from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from matplotlib.figure import Figure
import matplotlib.animation as animation

import Tkinter as tk
import ttk

a = Figure(figsize=(4,4))
plot_a = a.add_subplot(111)

b = Figure(figsize=(4,4))
plot_b = b.add_subplot(111)

x = [1,2,3,4,5]
y_a = [1,4,9,16,25]
y_b = [25,16,9,4,1]


def updateGraphs(i):

    plot_a.clear()
    plot_a.plot(x,y_a)

    plot_b.clear()
    plot_b.plot(x,y_b)

class TransientAnalysis(tk.Tk):

    def __init__(self,*args,**kwargs):

        tk.Tk.__init__(self,*args,**kwargs)
        tk.Tk.wm_title(self, "Transient Analysis GUI: v1.0")

        container = tk.Frame(self)      
        container.pack(side="top", fill="both", expand=True)      
        container.grid_rowconfigure(0, weight=1)
        container.grid_columnconfigure(0, weight=1)

        self.frames = {}

        for F in ( GraphPageA, GraphPageB):

            frame = F(container, self)

            self.frames[F] = frame

            frame.grid(row=0, column=0, sticky="nsew")

        self.show_frame(GraphPageA)

    def show_frame(self, cont):

        frame = self.frames[cont]
        frame.tkraise()    

class GraphPageA(tk.Frame):

    def __init__(self, parent, controller):

        tk.Frame.__init__(self, parent)

        button1 = ttk.Button(self, text="Show Graph B",
                            command = lambda: controller.show_frame(GraphPageB))
        button1.grid(row=1, column=0,pady=20,padx=10, sticky='w')

        canvasA = FigureCanvasTkAgg(a, self)
        canvasA.show()
        canvasA.get_tk_widget().grid(row=1, column=1, pady=20,padx=10, sticky='nsew')

class GraphPageB(tk.Frame):

    def __init__(self, parent, controller):

        tk.Frame.__init__(self, parent)

        button1 = ttk.Button(self, text="Show Graph A",
                            command = lambda: controller.show_frame(GraphPageA))
        button1.grid(row=1, column=0,pady=20,padx=10, sticky='w')

        canvasB = FigureCanvasTkAgg(b, self)
        canvasB.show()
        canvasB.get_tk_widget().grid(row=1, column=1, pady=20,padx=10, sticky='nsew')

app = TransientAnalysis()
app.geometry("800x600")
aniA    = animation.FuncAnimation(a, updateGraphs, interval=1000)
aniB    = animation.FuncAnimation(b, updateGraphs, interval=1000)
app.mainloop()

Чтобы быть более конкретным, страница, которая правильно отображает график, является последней, вызываемой в цикле for for F in ( GraphPageA, GraphPageB):

Я также пытался использовать разные функции updateGraphs, по одной для каждого графика, но результат тот же.

Как я могу иметь две страницы TKinter, отображающие эти два разных графика? Я использую Python 2.7.


person Alexandre de Castro Maciel    schedule 16.11.2015    source источник
comment
Я забыл упомянуть. Питон 2.7.   -  person Alexandre de Castro Maciel    schedule 16.11.2015
comment
Вы можете отредактировать свои собственные вопросы, чтобы обновить такие детали.   -  person tacaswell    schedule 16.11.2015
comment
Ну, это сводит с ума... Кажется, плохое взаимодействие между обратными вызовами анимации и деталями Tk, которые вы используете для панели, поскольку проблема в том, что одна из анимаций никогда не вызывается.   -  person tacaswell    schedule 16.11.2015
comment
ааа, разобрался, отвечай коротко.   -  person tacaswell    schedule 16.11.2015


Ответы (1)


from __future__ import print_function
import matplotlib
import numpy as np

matplotlib.use("TkAgg")

from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from matplotlib.figure import Figure
import matplotlib.animation as animation

import Tkinter as tk
import ttk


a = Figure(figsize=(4, 4))
plot_a = a.add_subplot(111)
plot_a.set_xlim([0, 2*np.pi])
plot_a.set_ylim([-1, 1])
lna, = plot_a.plot([], [], color='orange', lw=5)

b = Figure(figsize=(4, 4))
plot_b = b.add_subplot(111)
plot_b.set_xlim([0, 2*np.pi])
plot_b.set_ylim([-1, 1])

lnb, = plot_b.plot([], [], color='olive', lw=5)

x = np.linspace(0, 2*np.pi, 1024)


def updateGraphsA(i):
    lna.set_xdata(x)
    lna.set_ydata(np.sin(x + i * np.pi / 10))
    print('in A')


def updateGraphsB(i):
    lnb.set_xdata(x)
    lnb.set_ydata(np.sin(x - i * np.pi / 10))

    print('in B')


class TransientAnalysis(tk.Tk):

    def __init__(self, *args, **kwargs):
        self._running_anim = None
        tk.Tk.__init__(self, *args, **kwargs)
        tk.Tk.wm_title(self, "Transient Analysis GUI: v1.0")

        container = tk.Frame(self)
        container.pack(side="top", fill="both", expand=True)
        container.grid_rowconfigure(0, weight=1)
        container.grid_columnconfigure(0, weight=1)

        self.frames = {}

        for F in (GraphPageA, GraphPageB):

            frame = F(container, self)

            self.frames[F] = frame

            frame.grid(row=0, column=0, sticky="nsew")

        self.show_frame(GraphPageA)

    def show_frame(self, cont):

        frame = self.frames[cont]
        frame.tkraise()
        frame.canvas.draw_idle()


class GraphPageA(tk.Frame):

    def __init__(self, parent, controller):

        tk.Frame.__init__(self, parent)

        button1 = ttk.Button(self, text="Show Graph B",
                             command=(
                                 lambda: controller.show_frame(GraphPageB)))
        button1.grid(row=1, column=0, pady=20, padx=10, sticky='w')

        canvasA = FigureCanvasTkAgg(a, self)
        canvasA.show()
        canvasA.get_tk_widget().grid(
            row=1, column=1, pady=20, padx=10, sticky='nsew')
        self.canvas = canvasA


class GraphPageB(tk.Frame):

    def __init__(self, parent, controller):

        tk.Frame.__init__(self, parent)

        button1 = ttk.Button(self, text="Show Graph A",
                             command=(
                                 lambda: controller.show_frame(GraphPageA)))

        button1.grid(row=1, column=0, pady=20, padx=10, sticky='w')

        canvasB = FigureCanvasTkAgg(b, self)
        canvasB.show()
        canvasB.get_tk_widget().grid(
            row=1, column=1, pady=20, padx=10, sticky='nsew')
        self.canvas = canvasB


app = TransientAnalysis()
app.geometry("800x600")
aniA = animation.FuncAnimation(a, updateGraphsA, interval=1000, blit=False)
aniB = animation.FuncAnimation(b, updateGraphsB, interval=1000, blit=False)

app.mainloop()

Проблема заключалась в том, что анимация запускается при первом рисовании фигуры. По какой-либо причине (которая будет какой-то неприятной внутренней деталью либо Tk, либо mpl) событие события рисования для не видимых изначально осей никогда не запускается (или запускается до создания анимации ), поэтому вторая анимация даже не запускается. Добавляя атрибут canvas к своим классам, вы можете явно активировать draw_idle каждый раз, когда вы переключаете отображаемый график.

Обратите внимание, что я также:

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

На более общем уровне вам, вероятно, следует реорганизовать свои классы GraphPage*, чтобы они были одним параметризованным классом, который отвечает за:

  • создание и удержание своих художников
  • создание и удержание объекта анимации
  • переключать показанный график на основе строкового ключа, а не типа класса.

Например:

from __future__ import print_function
import matplotlib
import numpy as np

matplotlib.use("TkAgg")

from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from matplotlib.figure import Figure
import matplotlib.animation as animation

import Tkinter as tk
import ttk

x = np.linspace(0, 2*np.pi, 1024)


class TransientAnalysis(tk.Tk):

    pages = ((1, 'Switch to "-"', '-', '+', 'orange'),
             (-1, 'Switch to "+"', '+', '-', 'olive'))

    def __init__(self, *args, **kwargs):
        self._running_anim = None
        tk.Tk.__init__(self, *args, **kwargs)
        tk.Tk.wm_title(self, "Transient Analysis GUI: v1.0")

        container = tk.Frame(self)
        container.pack(side="top", fill="both", expand=True)
        container.grid_rowconfigure(0, weight=1)
        container.grid_columnconfigure(0, weight=1)

        self.frames = {}

        for (direction, text, other_key, my_key, color) in self.pages:

            frame = MovingSinGraphPage(direction, text, other_key,
                                       my_key, color,
                                       container, self)

            self.frames[my_key] = frame

            frame.grid(row=0, column=0, sticky="nsew")

    def show_frame(self, cont):

        frame = self.frames[cont]
        frame.tkraise()
        frame.canvas.draw_idle()


class MovingSinGraphPage(tk.Frame):

    def __init__(self, move_dir, text, other_key, my_key,
                 color, parent, controller):
        self._sgn = np.sign(move_dir)

        tk.Frame.__init__(self, parent)

        button1 = ttk.Button(self, text=text,
                             command=(
                                 lambda: controller.show_frame(other_key)))
        button1.grid(row=1, column=0, pady=20, padx=10, sticky='w')

        # make mpl objects
        a = Figure(figsize=(4, 4))
        plot_a = a.add_subplot(111)
        # set up the axes limits and title
        plot_a.set_title(my_key)
        plot_a.set_xlim([0, 2*np.pi])
        plot_a.set_ylim([-1, 1])
        # make and stash the plot artist
        lna, = plot_a.plot([], [], color=color, lw=5)
        self._line = lna

        # make the canvas to integrate with Tk
        canvasA = FigureCanvasTkAgg(a, self)
        canvasA.show()
        canvasA.get_tk_widget().grid(
            row=1, column=1, pady=20, padx=10, sticky='nsew')

        # stash the canvas so that we can use it above to ensure a re-draw
        # when we switch to this page
        self.canvas = canvasA
        # create and save the animation
        self.anim = animation.FuncAnimation(a, self.update,
                                            interval=100)

    def update(self, i):
        self._line.set_xdata(x)
        self._line.set_ydata(np.sin(x + self._sgn * i * np.pi / 10))


app = TransientAnalysis()
app.geometry("800x600")

app.mainloop()

Это, видимо, привлекло мое внимание сегодня ..

person tacaswell    schedule 16.11.2015
comment
Что касается последнего кода, который вы разместили. Я получил следующую ошибку: frame = self.frames[cont] KeyError: 'orange' - person Alexandre de Castro Maciel; 16.11.2015
comment
Удалите строку self.show_frame в __init__ приложения верхнего уровня. - person tacaswell; 16.11.2015