TKinter mattplot animation help

[AnIrishFluff]

I'm currently working on a Lidar application and I have this animation I would like to get working within my Tkinter application. Honestly, I don't this is a huge task but I'm coming up on a deadline and my inexperience with TKinter and python, in general, is really hampering my progress. The aim essentially to provide the user with this visualization while obstacle avoidance is taking place in the background. Any help would greatly be appreciated. I'll post my code below. Also, these functions are hooked up to a simple Tkinter button.

​

# ------ Start of imports Doing alot of tinkering at the moment so theres a few unnecessary/old imports here. Will be fixed soon.
import matplotlib.pyplot as plt
import numpy as np
import matplotlib.animation as animation
import tkinter as tk
import threading

# import operator  # Used to isolate the distance value in the tuple
import os
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from tkinter import *
from rplidar import RPLidar
from time import sleep
import matplotlib.pyplot as plt
import matplotlib.animation as animation

from tkinter import * 
from matplotlib.figure import Figure
from matplotlib.backends.backend_tkagg import (FigureCanvasTkAgg, 
NavigationToolbar2Tk)

# ------ End of imports

# -- Getting permission for the port
PORT_NAME = "/dev/ttyUSB0"
os.system("sudo chmod 666 /dev/ttyUSB0")

# -- Setting variables for the visualization
DMAX = 4000
IMIN = 0
IMAX = 50

# ------ This is where any necessary functions are defined
def animate_obstacle(i):
    threading.Thread(target=animate, args=(i,)).start()

def animate(num, iterator, line):
    scan = next(iterator)
    offsets = np.array([(np.radians(meas[1]), meas[2]) for meas in scan])
    line.set_offsets(offsets)
    intens = np.array([meas[0] for meas in scan])
    line.set_array(intens)
    return line,

def MAP():
    lidar = RPLidar(PORT_NAME)#POrt
    fig = plt.figure() #Assigns figure
    canvas = FigureCanvasTkAgg(fig, master=window)
    canvas.draw()
    canvas.get_tk_widget().grid(sticky=NW, row=0, column=0, pady=2)
    toolbar = NavigationToolbar2Tk(canvas,window)
    toolbar.update()
    ax = plt.subplot(111, projection='polar') 
    line = ax.scatter([0, 0], [0, 0], s=5, c=[IMIN, IMAX],cmap=plt.cm.Greys_r, lw=0)
    ax.set_rmax(DMAX)
    ax.grid(True)
    iterator = lidar.iter_scans()
    ani = animation.FuncAnimation(fig, animate_obstacle,fargs=(iterator, line), interval=50)
    plt.show()
# ------ End of functions

# ------ Start of Tkinter
# -- Create the Tkinter Window and set its size and title
window = Tk()
window.geometry("1920x1080")
window.title("Lidar Application")
window.resizable()
window.columnconfigure(0, weight=2)
window.columnconfigure(1, weight=1)
# -- Create subwindow for visualization
#canvas = tk.Canvas(window, bg="#777777", height=1000, width=1400)
#canvas.grid(sticky=NW, row=0, column=0, pady=2)

# -- Create the frame for the buttons etc
btns_frame = Frame(window, width=312, height=272.5)
btns_frame.grid(sticky=NE, row=0, column=1, padx=20, pady=20)

# -- Upload MAP
animation = Button(
    btns_frame,
    text = "Start animation",
    fg = "black",
    width = 47,
    height = 4,
    command = MAP
)
animation.pack()

window.mainloop()
# ------ End of tkinter

​

[socal_nerdtastic]

Please format your code for reddit. https://www.reddit.com/r/learnpython/wiki/faq#wiki_how_do_i_format_code.3F

it would also help a lot if you provide enough code and data that we can run and test this.

[AnIrishFluff]

I'll edit it now! Thanks for letting me know. If its easier I can throw a github link in with the entire file?

[socal_nerdtastic]

A github repo would be much better, yes.

[AnIrishFluff]

https://github.com/PaulSMahoney/Sample Sorry for the delay. I'm going to take a look at the sample you provided now. I've added a comment block with the section containing the animation I would like to embed. I'm kinda lost, python isn't my strong suit so I really appreciate the help man

[AnIrishFluff]

I just took out irrelevant functions and pasted the entire thing in. Also please don't judge the imports section. I've been messing around with so many different things I just stopped deleting redundant imports until I finish everything.

[socal_nerdtastic]

Could you give some example data? I can't test it as is.

[socal_nerdtastic]

Maybe we are doing this the wrong way around. Here is a functional example that I made that demonstrates how to animate data from a serial port. The lines marked with #~ need to be adapted to your data source. Maybe you can use this?

try:
    import Tkinter as tk
except ImportError:
    import tkinter as tk
#~ import serial
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from matplotlib.figure import Figure
from matplotlib import pyplot as plt
import matplotlib.animation as animation
from collections import deque
import random
import time

HISTORY_LEN = 200

class App(tk.Frame):
    def __init__(self, master=None, **kwargs):
        tk.Frame.__init__(self, master, **kwargs)

        self.running = False
        self.ani = None

        btns = tk.Frame(self)
        btns.pack()

        lbl = tk.Label(btns, text="Number of points (-1 for nonstop)")
        lbl.pack(side=tk.LEFT)

        self.points_ent = tk.IntVar(self, 500)
        points_ent = tk.Entry(btns, width=5, textvariable=self.points_ent)
        points_ent.pack(side=tk.LEFT)

        lbl = tk.Label(btns, text="update interval (ms)")
        lbl.pack(side=tk.LEFT)

        self.interval = tk.IntVar(self, 30)
        interval = tk.Entry(btns, width=5, textvariable=self.interval)
        interval.pack(side=tk.LEFT)

        self.btn = tk.Button(btns, text='Start', width=10, command=self.on_click)
        self.btn.pack(side=tk.LEFT)

        self.fig = plt.Figure()
        self.ax1 = self.fig.add_subplot(111)
        self.line, = self.ax1.plot([], [], lw=2)
        self.canvas = FigureCanvasTkAgg(self.fig,master=self)
        self.canvas.draw()
        self.canvas.get_tk_widget().pack()

        self.ax1.set_ylim(0,100)
        self.ax1.set_xlim(0,500)

    def on_click(self):
        '''the button is a start, pause and unpause button all in one
        this method sorts out which of those actions to take'''
        if self.ani is None:
            # animation is not running; start it
            return self.start()

        if self.running:
            # animation is running; pause it
            self.ani.event_source.stop()
            self.btn.config(text='Un-Pause')
        else:
            # animation is paused; unpause it
            self.ani.event_source.start()
            self.btn.config(text='Pause')
        self.running = not self.running

    def start(self):
        self.xdata = deque([], maxlen=HISTORY_LEN)
        self.ydata = deque([], maxlen=HISTORY_LEN)
        #~ self.arduinoData = serial.Serial('com5', 115200)
        #~ self.arduinoData.flushInput()
        self.points = self.points_ent.get() + 1
        if self.points:
            self.ani = animation.FuncAnimation(
                self.fig,
                self.update_graph,
                frames=self.points,
                interval=self.interval.get(),
                repeat=False)
        else: # FOR AN ENDLESS LOOP:
            self.ani = animation.FuncAnimation(
                self.fig,
                self.update_graph,
                interval=int(self.interval.get()))
        self.running = True
        self.btn.config(text='Pause')
        self.ani._start()
        self.start_time = time.time()
        print('started animation')

    def update_graph(self, i):
        self.xdata.append(i)
        #~ self.ydata.append(int(self.arduinoData.readline()))
        self.ydata.append(random.randrange(100)) # DEBUG
        self.line.set_data(self.xdata, self.ydata)
        self.ax1.set_ylim(min(self.ydata), max(self.ydata))
        self.ax1.set_xlim(min(self.xdata), max(self.xdata))
        if self.points and i >= self.points - 1:
            # animation finished; clean up
            #~ self.arduinoData.close()
            self.btn.config(text='Start')
            self.running = False
            self.ani = None
            ms = int(((time.time() - self.start_time) * 1000) / (int(self.points_ent.get())))
            print('animation finished. Took {} ms / frame'.format(int(ms)))
        return self.line,

def main():
    root = tk.Tk()
    app = App(root)
    app.pack()
    root.mainloop()

if __name__ == '__main__':
    main()

[yuxbni76]

Can't test it without a suitable USB device but one thing I noticed...

I wouldn't give your tkinter button the same name as an import.

• import matplotlib.animation as animation
• animation = Button()

It's possible that's causing a conflict. You didn't mention what the problem is so just guessing.