##// END OF EJS Templates
schain
RSS
Fix version
Fix version
Juan C. Espinoza - - Load All Authors

File last commit:

r1327:e2336f44bb45
r1332:7f771b904961
Show More
jroplot_base.py
713 lines | 24.0 KiB | text/x-python | PythonLexer
/ schainpy / model / graphics / jroplot_base.py
History | Source | Raw |Copy content |Copy permalink
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187
import os
import sys
import zmq
import time
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 import numpy
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 import datetime
Juan C. Espinoza
Update sender plots when loosing conn
 r1302 from queue import Queue
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 from functools import wraps
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 from threading import Thread
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 import matplotlib
if 'BACKEND' in os.environ:
matplotlib.use(os.environ['BACKEND'])
elif 'linux' in sys.platform:
matplotlib.use("TkAgg")
elif 'darwin' in sys.platform:
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 matplotlib.use('WxAgg')
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 else:
from schainpy.utils import log
log.warning('Using default Backend="Agg"', 'INFO')
matplotlib.use('Agg')
import matplotlib.pyplot as plt
from matplotlib.patches import Polygon
from mpl_toolkits.axes_grid1 import make_axes_locatable
from matplotlib.ticker import FuncFormatter, LinearLocator, MultipleLocator
from schainpy.model.data.jrodata import PlotterData
from schainpy.model.proc.jroproc_base import ProcessingUnit, Operation, MPDecorator
from schainpy.utils import log
jet_values = matplotlib.pyplot.get_cmap('jet', 100)(numpy.arange(100))[10:90]
blu_values = matplotlib.pyplot.get_cmap(
'seismic_r', 20)(numpy.arange(20))[10:15]
ncmap = matplotlib.colors.LinearSegmentedColormap.from_list(
'jro', numpy.vstack((blu_values, jet_values)))
matplotlib.pyplot.register_cmap(cmap=ncmap)
CMAPS = [plt.get_cmap(s) for s in ('jro', 'jet', 'viridis',
'plasma', 'inferno', 'Greys', 'seismic', 'bwr', 'coolwarm')]
EARTH_RADIUS = 6.3710e3
def ll2xy(lat1, lon1, lat2, lon2):
p = 0.017453292519943295
a = 0.5 - numpy.cos((lat2 - lat1) * p)/2 + numpy.cos(lat1 * p) * \
numpy.cos(lat2 * p) * (1 - numpy.cos((lon2 - lon1) * p)) / 2
r = 12742 * numpy.arcsin(numpy.sqrt(a))
theta = numpy.arctan2(numpy.sin((lon2-lon1)*p)*numpy.cos(lat2*p), numpy.cos(lat1*p)
* numpy.sin(lat2*p)-numpy.sin(lat1*p)*numpy.cos(lat2*p)*numpy.cos((lon2-lon1)*p))
theta = -theta + numpy.pi/2
return r*numpy.cos(theta), r*numpy.sin(theta)
def km2deg(km):
'''
Convert distance in km to degrees
'''
return numpy.rad2deg(km/EARTH_RADIUS)
def figpause(interval):
backend = plt.rcParams['backend']
if backend in matplotlib.rcsetup.interactive_bk:
figManager = matplotlib._pylab_helpers.Gcf.get_active()
if figManager is not None:
canvas = figManager.canvas
if canvas.figure.stale:
canvas.draw()
try:
canvas.start_event_loop(interval)
except:
pass
return
def popup(message):
'''
'''
fig = plt.figure(figsize=(12, 8), facecolor='r')
text = '\n'.join([s.strip() for s in message.split(':')])
fig.text(0.01, 0.5, text, ha='left', va='center',
size='20', weight='heavy', color='w')
fig.show()
figpause(1000)
class Throttle(object):
'''
Decorator that prevents a function from being called more than once every
time period.
To create a function that cannot be called more than once a minute, but
will sleep until it can be called:
@Throttle(minutes=1)
def foo():
pass
for i in range(10):
foo()
print "This function has run %s times." % i
'''
def __init__(self, seconds=0, minutes=0, hours=0):
self.throttle_period = datetime.timedelta(
seconds=seconds, minutes=minutes, hours=hours
)
self.time_of_last_call = datetime.datetime.min
def __call__(self, fn):
@wraps(fn)
def wrapper(*args, **kwargs):
coerce = kwargs.pop('coerce', None)
if coerce:
self.time_of_last_call = datetime.datetime.now()
return fn(*args, **kwargs)
else:
now = datetime.datetime.now()
time_since_last_call = now - self.time_of_last_call
time_left = self.throttle_period - time_since_last_call
if time_left > datetime.timedelta(seconds=0):
return
self.time_of_last_call = datetime.datetime.now()
return fn(*args, **kwargs)
return wrapper
def apply_throttle(value):
@Throttle(seconds=value)
def fnThrottled(fn):
fn()
return fnThrottled
@MPDecorator
class Plot(Operation):
'''
Base class for Schain plotting operations
'''
CODE = 'Figure'
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 colormap = 'jet'
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 bgcolor = 'white'
Juan C. Espinoza
Update and fix plot modules #TODO: correlation & spectraheis
 r1285 buffering = True
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 __missing = 1E30
Juan C. Espinoza
Update and fix plot modules #TODO: correlation & spectraheis
 r1285 __attrs__ = ['show', 'save', 'ymin', 'ymax', 'zmin', 'zmax', 'title',
'showprofile']
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187
def __init__(self):
Operation.__init__(self)
self.isConfig = False
self.isPlotConfig = False
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 self.save_counter = 1
Juan C. Espinoza
Fix sending data to server in plots
 r1299 self.sender_time = 0
Juan C. Espinoza
test for localtime plotting issues
 r1218 self.data = None
Juan C. Espinoza
Update and fix plot modules #TODO: correlation & spectraheis
 r1285 self.firsttime = True
Juan C. Espinoza
Correct localtime in plots, handle exception in send_to_server
 r1327 self.sender_queue = Queue(maxsize=60)
Juan C. Espinoza
Update and fix plot modules #TODO: correlation & spectraheis
 r1285 self.plots_adjust = {'left': 0.125, 'right': 0.9, 'bottom': 0.15, 'top': 0.9, 'wspace': 0.2, 'hspace': 0.2}
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187
def __fmtTime(self, x, pos):
'''
'''
return '{}'.format(self.getDateTime(x).strftime('%H:%M'))
def __setup(self, **kwargs):
'''
Initialize variables
'''
self.figures = []
self.axes = []
self.cb_axes = []
self.localtime = kwargs.pop('localtime', True)
self.show = kwargs.get('show', True)
self.save = kwargs.get('save', False)
jespinoza
Cleaning IO modules (base, voltaje and spectra)
 r1251 self.save_period = kwargs.get('save_period', 1)
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 self.colormap = kwargs.get('colormap', self.colormap)
self.colormap_coh = kwargs.get('colormap_coh', 'jet')
self.colormap_phase = kwargs.get('colormap_phase', 'RdBu_r')
self.colormaps = kwargs.get('colormaps', None)
self.bgcolor = kwargs.get('bgcolor', self.bgcolor)
self.showprofile = kwargs.get('showprofile', False)
self.title = kwargs.get('wintitle', self.CODE.upper())
self.cb_label = kwargs.get('cb_label', None)
self.cb_labels = kwargs.get('cb_labels', None)
self.labels = kwargs.get('labels', None)
self.xaxis = kwargs.get('xaxis', 'frequency')
self.zmin = kwargs.get('zmin', None)
self.zmax = kwargs.get('zmax', None)
self.zlimits = kwargs.get('zlimits', None)
self.xmin = kwargs.get('xmin', None)
self.xmax = kwargs.get('xmax', None)
Juan C. Espinoza
Update plots modules
 r1308 self.xrange = kwargs.get('xrange', 12)
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 self.xscale = kwargs.get('xscale', None)
self.ymin = kwargs.get('ymin', None)
self.ymax = kwargs.get('ymax', None)
self.yscale = kwargs.get('yscale', None)
self.xlabel = kwargs.get('xlabel', None)
jespinoza
Optimize plots
 r1269 self.attr_time = kwargs.get('attr_time', 'utctime')
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 self.decimation = kwargs.get('decimation', None)
self.showSNR = kwargs.get('showSNR', False)
self.oneFigure = kwargs.get('oneFigure', True)
self.width = kwargs.get('width', None)
self.height = kwargs.get('height', None)
self.colorbar = kwargs.get('colorbar', True)
self.factors = kwargs.get('factors', [1, 1, 1, 1, 1, 1, 1, 1])
self.channels = kwargs.get('channels', None)
self.titles = kwargs.get('titles', [])
self.polar = False
George Yong
Add ScopePlot to the new way of plotting data (jroplot_data)
 r1202 self.type = kwargs.get('type', 'iq')
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 self.grid = kwargs.get('grid', False)
self.pause = kwargs.get('pause', False)
Juan C. Espinoza
Minor bugs
 r1278 self.save_code = kwargs.get('save_code', None)
self.throttle = kwargs.get('throttle', 0)
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 self.exp_code = kwargs.get('exp_code', None)
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 self.plot_server = kwargs.get('plot_server', False)
Juan C. Espinoza
Fix sending data to server in plots
 r1299 self.sender_period = kwargs.get('sender_period', 60)
Juan C. Espinoza
Update plots modules
 r1308 self.tag = kwargs.get('tag', '')
Juan C. Espinoza
Update and fix plot modules #TODO: correlation & spectraheis
 r1285 self.height_index = kwargs.get('height_index', None)
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 self.__throttle_plot = apply_throttle(self.throttle)
self.data = PlotterData(
Juan C. Espinoza
Correct localtime in plots, handle exception in send_to_server
 r1327 self.CODE, self.throttle, self.exp_code, self.localtime, self.buffering, snr=self.showSNR)
Juan C. Espinoza
Allow to send data to new realtime app
 r1212
if self.plot_server:
if not self.plot_server.startswith('tcp://'):
self.plot_server = 'tcp://{}'.format(self.plot_server)
log.success(
'Sending to server: {}'.format(self.plot_server),
self.name
)
Juan C. Espinoza
Update plots for realtime app
 r1221 if 'plot_name' in kwargs:
self.plot_name = kwargs['plot_name']
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187
def __setup_plot(self):
'''
Common setup for all figures, here figures and axes are created
'''
self.setup()
Juan C. Espinoza
test for localtime plotting issues
 r1218 self.time_label = 'LT' if self.localtime else 'UTC'
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187
if self.width is None:
self.width = 8
self.figures = []
self.axes = []
self.cb_axes = []
self.pf_axes = []
self.cmaps = []
size = '15%' if self.ncols == 1 else '30%'
pad = '4%' if self.ncols == 1 else '8%'
if self.oneFigure:
if self.height is None:
self.height = 1.4 * self.nrows + 1
fig = plt.figure(figsize=(self.width, self.height),
edgecolor='k',
facecolor='w')
self.figures.append(fig)
for n in range(self.nplots):
ax = fig.add_subplot(self.nrows, self.ncols,
n + 1, polar=self.polar)
ax.tick_params(labelsize=8)
ax.firsttime = True
ax.index = 0
ax.press = None
self.axes.append(ax)
if self.showprofile:
cax = self.__add_axes(ax, size=size, pad=pad)
cax.tick_params(labelsize=8)
self.pf_axes.append(cax)
else:
if self.height is None:
self.height = 3
for n in range(self.nplots):
fig = plt.figure(figsize=(self.width, self.height),
edgecolor='k',
facecolor='w')
ax = fig.add_subplot(1, 1, 1, polar=self.polar)
ax.tick_params(labelsize=8)
ax.firsttime = True
ax.index = 0
ax.press = None
self.figures.append(fig)
self.axes.append(ax)
if self.showprofile:
cax = self.__add_axes(ax, size=size, pad=pad)
cax.tick_params(labelsize=8)
self.pf_axes.append(cax)
for n in range(self.nrows):
if self.colormaps is not None:
cmap = plt.get_cmap(self.colormaps[n])
else:
cmap = plt.get_cmap(self.colormap)
cmap.set_bad(self.bgcolor, 1.)
Juan C. Espinoza
Update and fix plot modules #TODO: correlation & spectraheis
 r1285 self.cmaps.append(cmap)
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187
def __add_axes(self, ax, size='30%', pad='8%'):
'''
Add new axes to the given figure
'''
divider = make_axes_locatable(ax)
nax = divider.new_horizontal(size=size, pad=pad)
ax.figure.add_axes(nax)
return nax
def fill_gaps(self, x_buffer, y_buffer, z_buffer):
'''
Create a masked array for missing data
'''
if x_buffer.shape[0] < 2:
return x_buffer, y_buffer, z_buffer
deltas = x_buffer[1:] - x_buffer[0:-1]
x_median = numpy.median(deltas)
index = numpy.where(deltas > 5 * x_median)
if len(index[0]) != 0:
z_buffer[::, index[0], ::] = self.__missing
z_buffer = numpy.ma.masked_inside(z_buffer,
0.99 * self.__missing,
1.01 * self.__missing)
return x_buffer, y_buffer, z_buffer
def decimate(self):
# dx = int(len(self.x)/self.__MAXNUMX) + 1
dy = int(len(self.y) / self.decimation) + 1
# x = self.x[::dx]
x = self.x
y = self.y[::dy]
z = self.z[::, ::, ::dy]
return x, y, z
def format(self):
'''
Set min and max values, labels, ticks and titles
'''
if self.xmin is None:
xmin = self.data.min_time
else:
if self.xaxis is 'time':
dt = self.getDateTime(self.data.min_time)
xmin = (dt.replace(hour=int(self.xmin), minute=0, second=0) -
datetime.datetime(1970, 1, 1)).total_seconds()
if self.data.localtime:
xmin += time.timezone
else:
xmin = self.xmin
if self.xmax is None:
xmax = xmin + self.xrange * 60 * 60
else:
if self.xaxis is 'time':
dt = self.getDateTime(self.data.max_time)
jespinoza
Optimize plots
 r1269 xmax = self.xmax - 1
xmax = (dt.replace(hour=int(xmax), minute=59, second=59) -
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 datetime.datetime(1970, 1, 1) + datetime.timedelta(seconds=1)).total_seconds()
if self.data.localtime:
xmax += time.timezone
else:
xmax = self.xmax
George Yong
Add ScopePlot to the new way of plotting data (jroplot_data)
 r1202
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 ymin = self.ymin if self.ymin else numpy.nanmin(self.y)
ymax = self.ymax if self.ymax else numpy.nanmax(self.y)
George Yong
Fix dynamic 'xstep' for all Plots
 r1204
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 for n, ax in enumerate(self.axes):
if ax.firsttime:
jespinoza
Optimize plots
 r1269
dig = int(numpy.log10(ymax))
if dig == 0:
digD = len(str(ymax)) - 2
ydec = ymax*(10**digD)
dig = int(numpy.log10(ydec))
ystep = ((ydec + (10**(dig)))//10**(dig))*(10**(dig))
ystep = ystep/5
ystep = ystep/(10**digD)
else:
ystep = ((ymax + (10**(dig)))//10**(dig))*(10**(dig))
ystep = ystep/5
if self.xaxis is not 'time':
dig = int(numpy.log10(xmax))
if dig <= 0:
digD = len(str(xmax)) - 2
xdec = xmax*(10**digD)
dig = int(numpy.log10(xdec))
xstep = ((xdec + (10**(dig)))//10**(dig))*(10**(dig))
xstep = xstep*0.5
xstep = xstep/(10**digD)
else:
xstep = ((xmax + (10**(dig)))//10**(dig))*(10**(dig))
xstep = xstep/5
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 ax.set_facecolor(self.bgcolor)
ax.yaxis.set_major_locator(MultipleLocator(ystep))
if self.xscale:
ax.xaxis.set_major_formatter(FuncFormatter(
lambda x, pos: '{0:g}'.format(x*self.xscale)))
if self.xscale:
ax.yaxis.set_major_formatter(FuncFormatter(
lambda x, pos: '{0:g}'.format(x*self.yscale)))
if self.xaxis is 'time':
ax.xaxis.set_major_formatter(FuncFormatter(self.__fmtTime))
ax.xaxis.set_major_locator(LinearLocator(9))
else:
ax.xaxis.set_major_locator(MultipleLocator(xstep))
if self.xlabel is not None:
ax.set_xlabel(self.xlabel)
ax.set_ylabel(self.ylabel)
ax.firsttime = False
if self.showprofile:
self.pf_axes[n].set_ylim(ymin, ymax)
self.pf_axes[n].set_xlim(self.zmin, self.zmax)
self.pf_axes[n].set_xlabel('dB')
self.pf_axes[n].grid(b=True, axis='x')
[tick.set_visible(False)
for tick in self.pf_axes[n].get_yticklabels()]
if self.colorbar:
ax.cbar = plt.colorbar(
ax.plt, ax=ax, fraction=0.05, pad=0.02, aspect=10)
ax.cbar.ax.tick_params(labelsize=8)
ax.cbar.ax.press = None
if self.cb_label:
ax.cbar.set_label(self.cb_label, size=8)
elif self.cb_labels:
ax.cbar.set_label(self.cb_labels[n], size=8)
else:
ax.cbar = None
if self.grid:
ax.grid(True)
if not self.polar:
ax.set_xlim(xmin, xmax)
ax.set_ylim(ymin, ymax)
ax.set_title('{} {} {}'.format(
self.titles[n],
self.getDateTime(self.data.max_time).strftime(
Juan C. Espinoza
Fix plotting bug
 r1214 '%Y-%m-%d %H:%M:%S'),
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 self.time_label),
size=8)
else:
ax.set_title('{}'.format(self.titles[n]), size=8)
ax.set_ylim(0, 90)
ax.set_yticks(numpy.arange(0, 90, 20))
ax.yaxis.labelpad = 40
Juan C. Espinoza
Update and fix plot modules #TODO: correlation & spectraheis
 r1285 if self.firsttime:
for n, fig in enumerate(self.figures):
fig.subplots_adjust(**self.plots_adjust)
self.firsttime = False
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 def clear_figures(self):
'''
Reset axes for redraw plots
'''
Juan C. Espinoza
Update and fix plot modules #TODO: correlation & spectraheis
 r1285 for ax in self.axes+self.pf_axes+self.cb_axes:
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 ax.clear()
ax.firsttime = True
Juan C. Espinoza
Correct localtime in plots, handle exception in send_to_server
 r1327 if hasattr(ax, 'cbar') and ax.cbar:
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 ax.cbar.remove()
def __plot(self):
'''
Main function to plot, format and save figures
'''
Juan C. Espinoza
Update and fix plot modules #TODO: correlation & spectraheis
 r1285 self.plot()
self.format()
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 for n, fig in enumerate(self.figures):
if self.nrows == 0 or self.nplots == 0:
log.warning('No data', self.name)
fig.text(0.5, 0.5, 'No Data', fontsize='large', ha='center')
fig.canvas.manager.set_window_title(self.CODE)
continue
Juan C. Espinoza
Update and fix plot modules #TODO: correlation & spectraheis
 r1285
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 fig.canvas.manager.set_window_title('{} - {}'.format(self.title,
self.getDateTime(self.data.max_time).strftime('%Y/%m/%d')))
fig.canvas.draw()
Juan C. Espinoza
Fix LT issue in plots add snr option in jrodata
 r1219 if self.show:
fig.show()
Juan C. Espinoza
Update and fix plot modules #TODO: correlation & spectraheis
 r1285 figpause(0.01)
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187
if self.save:
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 self.save_figure(n)
if self.plot_server:
self.send_to_server()
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 def save_figure(self, n):
'''
'''
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 if self.save_counter < self.save_period:
self.save_counter += 1
return
self.save_counter = 1
fig = self.figures[n]
Juan C. Espinoza
Minor bugs
 r1278 if self.save_code:
if isinstance(self.save_code, str):
labels = [self.save_code for x in self.figures]
else:
labels = self.save_code
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 else:
Juan C. Espinoza
Minor bugs
 r1278 labels = [self.CODE for x in self.figures]
Juan C. Espinoza
Allow to send data to new realtime app
 r1212
figname = os.path.join(
self.save,
Juan C. Espinoza
Minor bugs
 r1278 labels[n],
'{}_{}.png'.format(
labels[n],
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 self.getDateTime(self.data.max_time).strftime(
'%Y%m%d_%H%M%S'
),
)
)
log.log('Saving figure: {}'.format(figname), self.name)
if not os.path.isdir(os.path.dirname(figname)):
os.makedirs(os.path.dirname(figname))
fig.savefig(figname)
Juan C. Espinoza
Minor bugs
 r1278 if self.throttle == 0:
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 figname = os.path.join(
self.save,
Juan C. Espinoza
Minor bugs
 r1278 '{}_{}.png'.format(
labels[n],
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 self.getDateTime(self.data.min_time).strftime(
'%Y%m%d'
),
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 )
)
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 fig.savefig(figname)
def send_to_server(self):
'''
'''
Juan C. Espinoza
Update interval in send to server
 r1301 interval = self.data.tm - self.sender_time
if interval < self.sender_period:
Juan C. Espinoza
Update plot for moments pow, dop, width, snr
 r1229 return
Juan C. Espinoza
Allow to send data to new realtime app
 r1212
Juan C. Espinoza
Fix sending data to server in plots
 r1299 self.sender_time = self.data.tm
Juan C. Espinoza
Update plots modules
 r1308 attrs = ['titles', 'zmin', 'zmax', 'tag', 'ymin', 'ymax']
Juan C. Espinoza
Fix sending data to server in plots
 r1299 for attr in attrs:
Juan C. Espinoza
Update sender plots when loosing conn
 r1302 value = getattr(self, attr)
Juan C. Espinoza
Update plots modules
 r1308 if value:
if isinstance(value, (numpy.float32, numpy.float64)):
value = round(float(value), 2)
self.data.meta[attr] = value
if self.colormap == 'jet':
self.data.meta['colormap'] = 'Jet'
elif 'RdBu' in self.colormap:
self.data.meta['colormap'] = 'RdBu'
else:
self.data.meta['colormap'] = 'Viridis'
Juan C. Espinoza
Update interval in send to server
 r1301 self.data.meta['interval'] = int(interval)
Juan C. Espinoza
Update plots modules
 r1308 # msg = self.data.jsonify(self.data.tm, self.plot_name, self.plot_type)
Juan C. Espinoza
Correct localtime in plots, handle exception in send_to_server
 r1327 try:
self.sender_queue.put(self.data.tm, block=False)
except:
tm = self.sender_queue.get()
self.sender_queue.put(self.data.tm)
Juan C. Espinoza
Update sender plots when loosing conn
 r1302
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 while True:
Juan C. Espinoza
Update sender plots when loosing conn
 r1302 if self.sender_queue.empty():
break
Juan C. Espinoza
Update plots modules
 r1308 tm = self.sender_queue.get()
Juan C. Espinoza
Correct localtime in plots, handle exception in send_to_server
 r1327 try:
msg = self.data.jsonify(tm, self.plot_name, self.plot_type)
except:
continue
Juan C. Espinoza
Update plots modules
 r1308 self.socket.send_string(msg)
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 socks = dict(self.poll.poll(5000))
if socks.get(self.socket) == zmq.POLLIN:
reply = self.socket.recv_string()
if reply == 'ok':
log.log("Response from server ok", self.name)
Juan C. Espinoza
Correct localtime in plots, handle exception in send_to_server
 r1327 time.sleep(0.2)
Juan C. Espinoza
Update sender plots when loosing conn
 r1302 continue
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 else:
log.warning(
"Malformed reply from server: {}".format(reply), self.name)
else:
log.warning(
"No response from server, retrying...", self.name)
Juan C. Espinoza
Update plots modules
 r1308 self.sender_queue.put(self.data.tm)
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 self.socket.setsockopt(zmq.LINGER, 0)
self.socket.close()
self.poll.unregister(self.socket)
Juan C. Espinoza
Update sender plots when loosing conn
 r1302 time.sleep(0.1)
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 self.socket = self.context.socket(zmq.REQ)
self.socket.connect(self.plot_server)
self.poll.register(self.socket, zmq.POLLIN)
Juan C. Espinoza
Update sender plots when loosing conn
 r1302 break
Juan C. Espinoza
Allow to send data to new realtime app
 r1212
def setup(self):
'''
This method should be implemented in the child class, the following
attributes should be set:
self.nrows: number of rows
self.ncols: number of cols
self.nplots: number of plots (channels or pairs)
self.ylabel: label for Y axes
self.titles: list of axes title
'''
raise NotImplementedError
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187
def plot(self):
'''
Must be defined in the child class
'''
raise NotImplementedError
def run(self, dataOut, **kwargs):
Juan C. Espinoza
test for localtime plotting issues
 r1218 '''
Main plotting routine
'''
Juan C. Espinoza
Update and fix plot modules #TODO: correlation & spectraheis
 r1285
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 if self.isConfig is False:
self.__setup(**kwargs)
jespinoza
Optimize plots
 r1269
t = getattr(dataOut, self.attr_time)
Juan C. Espinoza
test for localtime plotting issues
 r1218
Juan C. Espinoza
Correct localtime in plots, handle exception in send_to_server
 r1327 if self.localtime:
Juan C. Espinoza
test for localtime plotting issues
 r1218 self.getDateTime = datetime.datetime.fromtimestamp
else:
self.getDateTime = datetime.datetime.utcfromtimestamp
Juan C. Espinoza
Update plots modules
 r1308 if self.xmin is None:
self.tmin = t
if 'buffer' in self.plot_type:
jespinoza
Optimize plots
 r1269 self.xmin = self.getDateTime(t).hour
Juan C. Espinoza
Update plots modules
 r1308 else:
self.tmin = (
self.getDateTime(t).replace(
hour=int(self.xmin),
minute=0,
second=0) - self.getDateTime(0)).total_seconds()
Juan C. Espinoza
test for localtime plotting issues
 r1218
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 self.data.setup()
self.isConfig = True
Juan C. Espinoza
Allow to send data to new realtime app
 r1212 if self.plot_server:
self.context = zmq.Context()
self.socket = self.context.socket(zmq.REQ)
self.socket.connect(self.plot_server)
self.poll = zmq.Poller()
self.poll.register(self.socket, zmq.POLLIN)
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187
jespinoza
Optimize plots
 r1269 tm = getattr(dataOut, self.attr_time)
Juan C. Espinoza
Update plots modules
 r1308 if self.data and (tm - self.tmin) >= self.xrange*60*60:
Juan C. Espinoza
Fix plotting bug
 r1214 self.save_counter = self.save_period
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 self.__plot()
Juan C. Espinoza
Update plots modules
 r1308 if 'time' in self.xaxis:
self.xmin += self.xrange
if self.xmin >= 24:
self.xmin -= 24
Juan C. Espinoza
Fix plotting bug
 r1214 self.tmin += self.xrange*60*60
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 self.data.setup()
self.clear_figures()
self.data.update(dataOut, tm)
if self.isPlotConfig is False:
self.__setup_plot()
self.isPlotConfig = True
Juan C. Espinoza
Minor bugs
 r1278 if self.throttle == 0:
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 self.__plot()
else:
Juan C. Espinoza
test for localtime plotting issues
 r1218 self.__throttle_plot(self.__plot)#, coerce=coerce)
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187
def close(self):
Juan C. Espinoza
Update README
 r1289 if self.data and not self.data.flagNoData:
Juan C. Espinoza
Fix plotting bug
 r1214 self.save_counter = self.save_period
self.__plot()
Juan C. Espinoza
Update README
 r1289 if self.data and not self.data.flagNoData and self.pause:
Juan C. Espinoza
New plotting architecture with buffering/throttle capabilities
 r1187 figpause(10)