Plotting data with mayavi via the subscriber node

edit

Hi, I solved this problem by instantiating a new window for plotting the data because the TkInter library requires that the GUI management has to be in main thread, and there are using the created window to plot the data in it.

the code for a new window:

#!/usr/bin/env python
import sys
from PyQt4.QtCore import *
from PyQt4.QtGui import *

import matplotlib
from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qt4agg import NavigationToolbar2QTAgg as  NavigationToolbar
from matplotlib.figure import Figure
from mpl_toolkits.mplot3d import Axes3D

class PlotWindow(QMainWindow):
    def __init__(self, parent=None):
        QMainWindow.__init__(self, parent)
        self.setWindowTitle('Sliding histogramm')
        self.create_main_frame()
        self.on_draw()

    def save_plot(self):
        pass

    def on_about(self):
        pass

    def on_pick(self, event):
        pass

    def on_draw(self):
        self.axes.clear()        
        self.axes.grid(True)
        self.canvas.draw()

    def create_main_frame(self):
        self.main_frame = QWidget()
        self.dpi = 100
        self.fig = Figure((5.0, 4.0), dpi=self.dpi)
        self.canvas = FigureCanvas(self.fig)
        self.canvas.setParent(self.main_frame)
        self.axes = self.fig.add_subplot(111, projection='3d')
        # self.axes.get_zaxis().set_visible(False)
        self.canvas.mpl_connect('pick_event', self.on_pick)
        self.mpl_toolbar = NavigationToolbar(self.canvas, self.main_frame)     
        vbox = QVBoxLayout()
        vbox.addWidget(self.canvas)
        vbox.addWidget(self.mpl_toolbar)
        self.main_frame.setLayout(vbox)
        self.setCentralWidget(self.main_frame)

and the code to plot:

#!/usr/bin/env python
import rospy
from sensor_msgs.msg import LaserScan
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
from windowonline import PlotWindow
from PyQt4.QtCore import *
from PyQt4.QtGui import *
import sys

class plotnew(PlotWindow): 
    def __init__(self): 
        PlotWindow.__init__(self)
        self.window_size=20
        rospy.init_node('listener', anonymous=True)
        self.subscriber=rospy.Subscriber("scan", LaserScan, self.callback)


    def callback(self, data):    
        print data
        theta = np.arange(data.angle_min, data.angle_max+data.angle_increment, data.angle_increment)
        z=np.ones(len(theta))
        self.axes.plot(theta, data.ranges,1,color='red',linewidth=2)
        self.canvas.draw()
        plt.pause(0.001)
        self.axes.clear()

if __name__ == '__main__':
    app = QApplication(sys.argv)
    window = plotnew()
    window.show()
    app.exec_()

Hope this helps someone who's having the same problem.

cheers, atom.