def add_cursor(self, x, y, **kwargs):
kwargs=process_kwargs(self, kwargs)
self.remove_collection()
self.clt=dict(h_line=self.plotter.axes.axhline(y, **kwargs),
v_line=self.plotter.axes.axvline(x, **kwargs),
v_axe=self.plotter.vert_axe.axhline(y, **kwargs),
h_axe=self.plotter.horiz_axe.axvline(x, **kwargs))
def scatter_plot(self, *args, **kwargs):
kwargs=process_kwargs(self, kwargs)
self.remove_collection()
if len(args)==1:
self.ydata=args[0]
self.xdata=arange(len(self.ydata))
elif len(args)==2:
self.xdata=args[0]
self.ydata=args[1]
self.clt=self.plotter.axes.scatter(self.xdata, self.ydata, **kwargs)
self.do_autolim()
def line_plot(self, *args, **kwargs):
kwargs=process_kwargs(self, kwargs)
self.remove_collection()
if len(args)==1:
y=args[0]
x=arange(len(y))
elif len(args)==2:
x=args[0]
y=args[1]
self.xdata=x
self.ydata=y
self.clt=self.plotter.axes.plot(*args, **kwargs)[0]
self.do_autolim()
def pcolormesh(self, *args, **kwargs):
kwargs=process_kwargs(self, kwargs)
self.remove_collection()
if len(args) == 1:
#if isinstance(args[0], tuple):
# self.zdata=zeros(args[0])
#else:
self.zdata=asanyarray(args[0])
numRows, numCols = self.zdata.shape
self.xdata=arange(numCols)
self.ydata=arange(numRows)
#elif len(args)==2:
# args=args+(zeros((len(self.xdata)-1, len(self.ydata)-1)),)
# self.xdata, self.ydata, self.zdata= [asanyarray(a) for a in args]
elif len(args) == 3:
self.xdata, self.ydata, self.zdata = [asanyarray(a) for a in args]
self.clt=self.plotter.axes.pcolormesh(self.xdata, self.ydata, self.zdata, **kwargs)
self.do_autolim()
def multiline_plot(self, *args, **kwargs):
kwargs=process_kwargs(self, kwargs)
self.xdata=[arg[0] for arg in args[0][0]]
self.zdata=[[vert[1] for vert in line] for line in args[0]]
self.clt=LineCollection(args[0])
self.plotter.axes.add_collection(self.clt)
def axhline(self, y, **kwargs):
kwargs=process_kwargs(self, kwargs)
self.remove_collection()
self.clt=self.plotter.axes.axhline(y, **kwargs)
def multiline(self, *args,**kwargs):
kwargs=process_kwargs(self, kwargs)
return multiline_plot(self, *args,**kwargs)
def scatter(self, *args, **kwargs):
kwargs=process_kwargs(self, kwargs)
return scatter_plot(self, *args, **kwargs)
def colormesh(self, *args, **kwargs):
kwargs=process_kwargs(self, kwargs)
return colormesh_plot(self, *args, **kwargs)
def hline(self, *args, **kwargs):
kwargs=process_kwargs(self, kwargs)
return hline_plot(self, *args, **kwargs)
def multiline(self, *args, **kwargs):
kwargs = process_kwargs(self, kwargs)
return multiline_plot(self, *args, **kwargs)
def colormesh(self, *args, **kwargs):
kwargs = process_kwargs(self, kwargs)
return colormesh_plot(self, *args, **kwargs)
def scatter(self, *args, **kwargs):
kwargs = process_kwargs(self, kwargs)
return scatter_plot(self, *args, **kwargs)
def hline(self, *args, **kwargs):
kwargs = process_kwargs(self, kwargs)
return hline_plot(self, *args, **kwargs)
