jroplot_data.py
783 lines
| 27.7 KiB
| text/x-python
|
PythonLexer
Juan C. Valdez
|
r865 | |||
| import os | ||||
| r889 | import zmq | |||
|
Juan C. Valdez
|
r865 | import time | ||
| import numpy | ||||
| import datetime | ||||
| import numpy as np | ||||
José Chávez
|
r927 | import matplotlib | ||
| matplotlib.use('TkAgg') | ||||
|
Juan C. Valdez
|
r865 | import matplotlib.pyplot as plt | ||
| from mpl_toolkits.axes_grid1 import make_axes_locatable | ||||
| from matplotlib.ticker import FuncFormatter, LinearLocator | ||||
| r889 | from multiprocessing import Process | |||
|
Juan C. Valdez
|
r865 | |||
| from schainpy.model.proc.jroproc_base import Operation | ||||
|
José Chávez
|
r933 | plt.ion() | ||
| r889 | ||||
| r922 | func = lambda x, pos: ('%s') %(datetime.datetime.fromtimestamp(x).strftime('%H:%M')) | |||
|
José Chávez
|
r1004 | fromtimestamp = lambda x, mintime : (datetime.datetime.utcfromtimestamp(mintime).replace(hour=(x + 5), minute=0) - d1970).total_seconds() | ||
|
Juan C. Valdez
|
r865 | |||
| d1970 = datetime.datetime(1970,1,1) | ||||
| r889 | class PlotData(Operation, Process): | |||
|
Juan C. Valdez
|
r865 | |||
| r889 | CODE = 'Figure' | |||
| r922 | colormap = 'jro' | |||
|
José Chávez
|
r931 | CONFLATE = False | ||
|
Juan C. Valdez
|
r865 | __MAXNUMX = 80 | ||
| __missing = 1E30 | ||||
| r889 | def __init__(self, **kwargs): | |||
|
Juan C. Valdez
|
r865 | |||
|
Juan C. Espinoza
|
r906 | Operation.__init__(self, plot=True, **kwargs) | ||
| r889 | Process.__init__(self) | |||
|
Juan C. Espinoza
|
r906 | self.kwargs['code'] = self.CODE | ||
| r889 | self.mp = False | |||
|
Juan C. Valdez
|
r865 | self.dataOut = None | ||
| self.isConfig = False | ||||
| self.figure = None | ||||
| r889 | self.axes = [] | |||
|
Juan C. Valdez
|
r865 | self.localtime = kwargs.pop('localtime', True) | ||
| r889 | self.show = kwargs.get('show', True) | |||
| self.save = kwargs.get('save', False) | ||||
| self.colormap = kwargs.get('colormap', self.colormap) | ||||
| r922 | self.colormap_coh = kwargs.get('colormap_coh', 'jet') | |||
| self.colormap_phase = kwargs.get('colormap_phase', 'RdBu_r') | ||||
| r907 | self.showprofile = kwargs.get('showprofile', True) | |||
|
Juan C. Valdez
|
r866 | self.title = kwargs.get('wintitle', '') | ||
| r922 | self.xaxis = kwargs.get('xaxis', 'frequency') | |||
|
Juan C. Valdez
|
r865 | self.zmin = kwargs.get('zmin', None) | ||
| self.zmax = kwargs.get('zmax', None) | ||||
| r889 | self.xmin = kwargs.get('xmin', None) | |||
| self.xmax = kwargs.get('xmax', None) | ||||
| self.xrange = kwargs.get('xrange', 24) | ||||
|
Juan C. Valdez
|
r866 | self.ymin = kwargs.get('ymin', None) | ||
| self.ymax = kwargs.get('ymax', None) | ||||
|
José Chávez
|
r1004 | self.__MAXNUMY = kwargs.get('decimation', 5000) | ||
| r916 | self.throttle_value = 5 | |||
|
José Chávez
|
r931 | self.times = [] | ||
|
José Chávez
|
r933 | #self.interactive = self.kwargs['parent'] | ||
| r922 | ||||
|
Juan C. Valdez
|
r865 | def fill_gaps(self, x_buffer, y_buffer, z_buffer): | ||
| if x_buffer.shape[0] < 2: | ||||
| return x_buffer, y_buffer, z_buffer | ||||
| deltas = x_buffer[1:] - x_buffer[0:-1] | ||||
| x_median = np.median(deltas) | ||||
| index = np.where(deltas > 5*x_median) | ||||
| if len(index[0]) != 0: | ||||
|
José Chávez
|
r897 | z_buffer[::, index[0], ::] = self.__missing | ||
|
Juan C. Valdez
|
r865 | z_buffer = np.ma.masked_inside(z_buffer, | ||
| 0.99*self.__missing, | ||||
| 1.01*self.__missing) | ||||
| return x_buffer, y_buffer, z_buffer | ||||
|
Juan C. Valdez
|
r866 | def decimate(self): | ||
| r889 | ||||
|
José Chávez
|
r898 | # dx = int(len(self.x)/self.__MAXNUMX) + 1 | ||
|
Juan C. Valdez
|
r866 | dy = int(len(self.y)/self.__MAXNUMY) + 1 | ||
| r889 | ||||
|
José Chávez
|
r898 | # x = self.x[::dx] | ||
| x = self.x | ||||
| r889 | y = self.y[::dy] | |||
|
José Chávez
|
r898 | z = self.z[::, ::, ::dy] | ||
| r889 | ||||
|
Juan C. Valdez
|
r866 | return x, y, z | ||
| r889 | def __plot(self): | |||
| print 'plotting...{}'.format(self.CODE) | ||||
|
Juan C. Valdez
|
r866 | |||
| r911 | if self.show: | |||
| self.figure.show() | ||||
|
Juan C. Valdez
|
r866 | self.plot() | ||
| r922 | plt.tight_layout() | |||
|
Juan C. Espinoza
|
r937 | self.figure.canvas.manager.set_window_title('{} {} - {}'.format(self.title, self.CODE.upper(), | ||
| datetime.datetime.fromtimestamp(self.max_time).strftime('%Y/%m/%d'))) | ||||
| r889 | ||||
|
Juan C. Valdez
|
r866 | if self.save: | ||
| r889 | figname = os.path.join(self.save, '{}_{}.png'.format(self.CODE, | |||
| r922 | datetime.datetime.fromtimestamp(self.saveTime).strftime('%y%m%d_%H%M%S'))) | |||
|
Juan C. Valdez
|
r866 | print 'Saving figure: {}'.format(figname) | ||
| self.figure.savefig(figname) | ||||
| self.figure.canvas.draw() | ||||
| r889 | def plot(self): | |||
| print 'plotting...{}'.format(self.CODE.upper()) | ||||
| return | ||||
|
Juan C. Valdez
|
r866 | |||
| r889 | def run(self): | |||
|
Juan C. Valdez
|
r866 | |||
| r889 | print '[Starting] {}'.format(self.name) | |||
|
Juan C. Espinoza
|
r937 | |||
| r889 | context = zmq.Context() | |||
| receiver = context.socket(zmq.SUB) | ||||
| receiver.setsockopt(zmq.SUBSCRIBE, '') | ||||
|
José Chávez
|
r897 | receiver.setsockopt(zmq.CONFLATE, self.CONFLATE) | ||
|
Juan C. Espinoza
|
r938 | |||
|
Juan C. Espinoza
|
r937 | if 'server' in self.kwargs['parent']: | ||
| receiver.connect('ipc:///tmp/{}.plots'.format(self.kwargs['parent']['server'])) | ||||
| else: | ||||
| receiver.connect("ipc:///tmp/zmq.plots") | ||||
|
Juan C. Valdez
|
r866 | |||
|
José Chávez
|
r931 | seconds_passed = 0 | ||
|
Juan C. Espinoza
|
r938 | |||
| r889 | while True: | |||
| try: | ||||
|
José Chávez
|
r931 | self.data = receiver.recv_pyobj(flags=zmq.NOBLOCK)#flags=zmq.NOBLOCK | ||
| self.started = self.data['STARTED'] | ||||
| r889 | self.dataOut = self.data['dataOut'] | |||
|
José Chávez
|
r931 | |||
| if (len(self.times) < len(self.data['times']) and not self.started and self.data['ENDED']): | ||||
| continue | ||||
| r889 | self.times = self.data['times'] | |||
| self.times.sort() | ||||
|
José Chávez
|
r898 | self.throttle_value = self.data['throttle'] | ||
| r889 | self.min_time = self.times[0] | |||
| self.max_time = self.times[-1] | ||||
|
Juan C. Valdez
|
r866 | |||
| r889 | if self.isConfig is False: | |||
|
José Chávez
|
r931 | print 'setting up' | ||
| r889 | self.setup() | |||
| self.isConfig = True | ||||
|
José Chávez
|
r931 | self.__plot() | ||
| r889 | ||||
|
José Chávez
|
r898 | if self.data['ENDED'] is True: | ||
|
José Chávez
|
r931 | print '********GRAPHIC ENDED********' | ||
| self.ended = True | ||||
|
José Chávez
|
r898 | self.isConfig = False | ||
|
José Chávez
|
r931 | self.__plot() | ||
| elif seconds_passed >= self.data['throttle']: | ||||
| print 'passed', seconds_passed | ||||
| self.__plot() | ||||
| seconds_passed = 0 | ||||
| r889 | ||||
| except zmq.Again as e: | ||||
| print 'Waiting for data...' | ||||
|
José Chávez
|
r931 | plt.pause(2) | ||
| seconds_passed += 2 | ||||
|
Juan C. Valdez
|
r866 | |||
| def close(self): | ||||
| if self.dataOut: | ||||
| r922 | self.__plot() | |||
| r889 | ||||
|
Juan C. Valdez
|
r866 | |||
| class PlotSpectraData(PlotData): | ||||
| r889 | CODE = 'spc' | |||
| colormap = 'jro' | ||||
|
José Chávez
|
r897 | CONFLATE = False | ||
| r922 | ||||
| r889 | def setup(self): | |||
| ncolspan = 1 | ||||
| colspan = 1 | ||||
| self.ncols = int(numpy.sqrt(self.dataOut.nChannels)+0.9) | ||||
| self.nrows = int(self.dataOut.nChannels*1./self.ncols + 0.9) | ||||
| self.width = 3.6*self.ncols | ||||
| self.height = 3.2*self.nrows | ||||
| if self.showprofile: | ||||
| ncolspan = 3 | ||||
| colspan = 2 | ||||
| self.width += 1.2*self.ncols | ||||
| self.ylabel = 'Range [Km]' | ||||
| self.titles = ['Channel {}'.format(x) for x in self.dataOut.channelList] | ||||
| if self.figure is None: | ||||
| self.figure = plt.figure(figsize=(self.width, self.height), | ||||
| edgecolor='k', | ||||
| facecolor='w') | ||||
| else: | ||||
| self.figure.clf() | ||||
| n = 0 | ||||
| for y in range(self.nrows): | ||||
| for x in range(self.ncols): | ||||
|
José Chávez
|
r897 | if n >= self.dataOut.nChannels: | ||
| r889 | break | |||
| ax = plt.subplot2grid((self.nrows, self.ncols*ncolspan), (y, x*ncolspan), 1, colspan) | ||||
| if self.showprofile: | ||||
| ax.ax_profile = plt.subplot2grid((self.nrows, self.ncols*ncolspan), (y, x*ncolspan+colspan), 1, 1) | ||||
| ax.firsttime = True | ||||
| self.axes.append(ax) | ||||
| n += 1 | ||||
|
Juan C. Valdez
|
r865 | def plot(self): | ||
| r889 | ||||
| if self.xaxis == "frequency": | ||||
| x = self.dataOut.getFreqRange(1)/1000. | ||||
| xlabel = "Frequency (kHz)" | ||||
| elif self.xaxis == "time": | ||||
| x = self.dataOut.getAcfRange(1) | ||||
| xlabel = "Time (ms)" | ||||
| else: | ||||
| x = self.dataOut.getVelRange(1) | ||||
| xlabel = "Velocity (m/s)" | ||||
| y = self.dataOut.getHeiRange() | ||||
| z = self.data[self.CODE] | ||||
| for n, ax in enumerate(self.axes): | ||||
| if ax.firsttime: | ||||
| self.xmax = self.xmax if self.xmax else np.nanmax(x) | ||||
| self.xmin = self.xmin if self.xmin else -self.xmax | ||||
| self.ymin = self.ymin if self.ymin else np.nanmin(y) | ||||
| self.ymax = self.ymax if self.ymax else np.nanmax(y) | ||||
| self.zmin = self.zmin if self.zmin else np.nanmin(z) | ||||
| self.zmax = self.zmax if self.zmax else np.nanmax(z) | ||||
| ax.plot = ax.pcolormesh(x, y, z[n].T, | ||||
| vmin=self.zmin, | ||||
| vmax=self.zmax, | ||||
| cmap=plt.get_cmap(self.colormap) | ||||
| ) | ||||
| divider = make_axes_locatable(ax) | ||||
| cax = divider.new_horizontal(size='3%', pad=0.05) | ||||
| self.figure.add_axes(cax) | ||||
| plt.colorbar(ax.plot, cax) | ||||
| ax.set_xlim(self.xmin, self.xmax) | ||||
| ax.set_ylim(self.ymin, self.ymax) | ||||
| ax.set_ylabel(self.ylabel) | ||||
| ax.set_xlabel(xlabel) | ||||
| ax.firsttime = False | ||||
| if self.showprofile: | ||||
| ax.plot_profile= ax.ax_profile.plot(self.data['rti'][self.max_time][n], y)[0] | ||||
| ax.ax_profile.set_xlim(self.zmin, self.zmax) | ||||
| ax.ax_profile.set_ylim(self.ymin, self.ymax) | ||||
| ax.ax_profile.set_xlabel('dB') | ||||
| ax.ax_profile.grid(b=True, axis='x') | ||||
| r892 | ax.plot_noise = ax.ax_profile.plot(numpy.repeat(self.data['noise'][self.max_time][n], len(y)), y, | |||
| color="k", linestyle="dashed", lw=2)[0] | ||||
| r889 | [tick.set_visible(False) for tick in ax.ax_profile.get_yticklabels()] | |||
| else: | ||||
| ax.plot.set_array(z[n].T.ravel()) | ||||
| if self.showprofile: | ||||
| ax.plot_profile.set_data(self.data['rti'][self.max_time][n], y) | ||||
| r892 | ax.plot_noise.set_data(numpy.repeat(self.data['noise'][self.max_time][n], len(y)), y) | |||
|
Juan C. Valdez
|
r866 | |||
| r892 | ax.set_title('{} - Noise: {:.2f} dB'.format(self.titles[n], self.data['noise'][self.max_time][n]), | |||
| size=8) | ||||
| r922 | self.saveTime = self.max_time | |||
| class PlotCrossSpectraData(PlotData): | ||||
| CODE = 'cspc' | ||||
| zmin_coh = None | ||||
| zmax_coh = None | ||||
| zmin_phase = None | ||||
| zmax_phase = None | ||||
| CONFLATE = False | ||||
| def setup(self): | ||||
| ncolspan = 1 | ||||
| colspan = 1 | ||||
| self.ncols = 2 | ||||
| self.nrows = self.dataOut.nPairs | ||||
| self.width = 3.6*self.ncols | ||||
| self.height = 3.2*self.nrows | ||||
| self.ylabel = 'Range [Km]' | ||||
| self.titles = ['Channel {}'.format(x) for x in self.dataOut.channelList] | ||||
| if self.figure is None: | ||||
| self.figure = plt.figure(figsize=(self.width, self.height), | ||||
| edgecolor='k', | ||||
| facecolor='w') | ||||
| else: | ||||
| self.figure.clf() | ||||
| for y in range(self.nrows): | ||||
| for x in range(self.ncols): | ||||
| ax = plt.subplot2grid((self.nrows, self.ncols), (y, x), 1, 1) | ||||
| ax.firsttime = True | ||||
| self.axes.append(ax) | ||||
| def plot(self): | ||||
| if self.xaxis == "frequency": | ||||
| x = self.dataOut.getFreqRange(1)/1000. | ||||
| xlabel = "Frequency (kHz)" | ||||
| elif self.xaxis == "time": | ||||
| x = self.dataOut.getAcfRange(1) | ||||
| xlabel = "Time (ms)" | ||||
| else: | ||||
| x = self.dataOut.getVelRange(1) | ||||
| xlabel = "Velocity (m/s)" | ||||
| y = self.dataOut.getHeiRange() | ||||
| z_coh = self.data['cspc_coh'] | ||||
| z_phase = self.data['cspc_phase'] | ||||
| for n in range(self.nrows): | ||||
| ax = self.axes[2*n] | ||||
| ax1 = self.axes[2*n+1] | ||||
| if ax.firsttime: | ||||
| self.xmax = self.xmax if self.xmax else np.nanmax(x) | ||||
| self.xmin = self.xmin if self.xmin else -self.xmax | ||||
| self.ymin = self.ymin if self.ymin else np.nanmin(y) | ||||
| self.ymax = self.ymax if self.ymax else np.nanmax(y) | ||||
| self.zmin_coh = self.zmin_coh if self.zmin_coh else 0.0 | ||||
| self.zmax_coh = self.zmax_coh if self.zmax_coh else 1.0 | ||||
| self.zmin_phase = self.zmin_phase if self.zmin_phase else -180 | ||||
| self.zmax_phase = self.zmax_phase if self.zmax_phase else 180 | ||||
| ax.plot = ax.pcolormesh(x, y, z_coh[n].T, | ||||
| vmin=self.zmin_coh, | ||||
| vmax=self.zmax_coh, | ||||
| cmap=plt.get_cmap(self.colormap_coh) | ||||
| ) | ||||
| divider = make_axes_locatable(ax) | ||||
| cax = divider.new_horizontal(size='3%', pad=0.05) | ||||
| self.figure.add_axes(cax) | ||||
| plt.colorbar(ax.plot, cax) | ||||
| ax.set_xlim(self.xmin, self.xmax) | ||||
| ax.set_ylim(self.ymin, self.ymax) | ||||
| r923 | ||||
| r922 | ax.set_ylabel(self.ylabel) | |||
| ax.set_xlabel(xlabel) | ||||
| ax.firsttime = False | ||||
| ax1.plot = ax1.pcolormesh(x, y, z_phase[n].T, | ||||
| vmin=self.zmin_phase, | ||||
| vmax=self.zmax_phase, | ||||
| cmap=plt.get_cmap(self.colormap_phase) | ||||
| ) | ||||
| divider = make_axes_locatable(ax1) | ||||
| cax = divider.new_horizontal(size='3%', pad=0.05) | ||||
| self.figure.add_axes(cax) | ||||
| plt.colorbar(ax1.plot, cax) | ||||
| ax1.set_xlim(self.xmin, self.xmax) | ||||
| ax1.set_ylim(self.ymin, self.ymax) | ||||
| ax1.set_ylabel(self.ylabel) | ||||
| ax1.set_xlabel(xlabel) | ||||
| ax1.firsttime = False | ||||
| else: | ||||
| ax.plot.set_array(z_coh[n].T.ravel()) | ||||
| ax1.plot.set_array(z_phase[n].T.ravel()) | ||||
| ax.set_title('Coherence Ch{} * Ch{}'.format(self.dataOut.pairsList[n][0], self.dataOut.pairsList[n][1]), size=8) | ||||
| ax1.set_title('Phase Ch{} * Ch{}'.format(self.dataOut.pairsList[n][0], self.dataOut.pairsList[n][1]), size=8) | ||||
| self.saveTime = self.max_time | ||||
| class PlotSpectraMeanData(PlotSpectraData): | ||||
| CODE = 'spc_mean' | ||||
| colormap = 'jet' | ||||
| def plot(self): | ||||
| if self.xaxis == "frequency": | ||||
| x = self.dataOut.getFreqRange(1)/1000. | ||||
| xlabel = "Frequency (kHz)" | ||||
| elif self.xaxis == "time": | ||||
| x = self.dataOut.getAcfRange(1) | ||||
| xlabel = "Time (ms)" | ||||
| else: | ||||
| x = self.dataOut.getVelRange(1) | ||||
| xlabel = "Velocity (m/s)" | ||||
| y = self.dataOut.getHeiRange() | ||||
| z = self.data['spc'] | ||||
| mean = self.data['mean'][self.max_time] | ||||
| for n, ax in enumerate(self.axes): | ||||
| if ax.firsttime: | ||||
| self.xmax = self.xmax if self.xmax else np.nanmax(x) | ||||
| self.xmin = self.xmin if self.xmin else -self.xmax | ||||
| self.ymin = self.ymin if self.ymin else np.nanmin(y) | ||||
| self.ymax = self.ymax if self.ymax else np.nanmax(y) | ||||
| self.zmin = self.zmin if self.zmin else np.nanmin(z) | ||||
| self.zmax = self.zmax if self.zmax else np.nanmax(z) | ||||
| ax.plt = ax.pcolormesh(x, y, z[n].T, | ||||
| vmin=self.zmin, | ||||
| vmax=self.zmax, | ||||
| cmap=plt.get_cmap(self.colormap) | ||||
| ) | ||||
| ax.plt_dop = ax.plot(mean[n], y, | ||||
| color='k')[0] | ||||
| divider = make_axes_locatable(ax) | ||||
| cax = divider.new_horizontal(size='3%', pad=0.05) | ||||
| self.figure.add_axes(cax) | ||||
| plt.colorbar(ax.plt, cax) | ||||
| ax.set_xlim(self.xmin, self.xmax) | ||||
| ax.set_ylim(self.ymin, self.ymax) | ||||
| ax.set_ylabel(self.ylabel) | ||||
| ax.set_xlabel(xlabel) | ||||
| ax.firsttime = False | ||||
| if self.showprofile: | ||||
| ax.plt_profile= ax.ax_profile.plot(self.data['rti'][self.max_time][n], y)[0] | ||||
| ax.ax_profile.set_xlim(self.zmin, self.zmax) | ||||
| ax.ax_profile.set_ylim(self.ymin, self.ymax) | ||||
| ax.ax_profile.set_xlabel('dB') | ||||
| ax.ax_profile.grid(b=True, axis='x') | ||||
| ax.plt_noise = ax.ax_profile.plot(numpy.repeat(self.data['noise'][self.max_time][n], len(y)), y, | ||||
| color="k", linestyle="dashed", lw=2)[0] | ||||
| [tick.set_visible(False) for tick in ax.ax_profile.get_yticklabels()] | ||||
| else: | ||||
| ax.plt.set_array(z[n].T.ravel()) | ||||
| ax.plt_dop.set_data(mean[n], y) | ||||
| if self.showprofile: | ||||
| ax.plt_profile.set_data(self.data['rti'][self.max_time][n], y) | ||||
| ax.plt_noise.set_data(numpy.repeat(self.data['noise'][self.max_time][n], len(y)), y) | ||||
| ax.set_title('{} - Noise: {:.2f} dB'.format(self.titles[n], self.data['noise'][self.max_time][n]), | ||||
| size=8) | ||||
| self.saveTime = self.max_time | ||||
|
Juan C. Valdez
|
r866 | |||
| class PlotRTIData(PlotData): | ||||
| r889 | ||||
| CODE = 'rti' | ||||
| colormap = 'jro' | ||||
| def setup(self): | ||||
|
Juan C. Valdez
|
r866 | self.ncols = 1 | ||
| self.nrows = self.dataOut.nChannels | ||||
| r889 | self.width = 10 | |||
| r922 | self.height = 2.2*self.nrows if self.nrows<6 else 12 | |||
| r907 | if self.nrows==1: | |||
| self.height += 1 | ||||
|
Juan C. Valdez
|
r866 | self.ylabel = 'Range [Km]' | ||
| r889 | self.titles = ['Channel {}'.format(x) for x in self.dataOut.channelList] | |||
| if self.figure is None: | ||||
| self.figure = plt.figure(figsize=(self.width, self.height), | ||||
| edgecolor='k', | ||||
| facecolor='w') | ||||
| else: | ||||
| self.figure.clf() | ||||
|
José Chávez
|
r898 | self.axes = [] | ||
| r889 | ||||
| for n in range(self.nrows): | ||||
| ax = self.figure.add_subplot(self.nrows, self.ncols, n+1) | ||||
| ax.firsttime = True | ||||
| self.axes.append(ax) | ||||
|
Juan C. Valdez
|
r866 | def plot(self): | ||
|
Juan C. Valdez
|
r865 | |||
| r889 | self.x = np.array(self.times) | |||
|
Juan C. Valdez
|
r866 | self.y = self.dataOut.getHeiRange() | ||
| self.z = [] | ||||
| r889 | ||||
| for ch in range(self.nrows): | ||||
| self.z.append([self.data[self.CODE][t][ch] for t in self.times]) | ||||
| self.z = np.array(self.z) | ||||
|
Juan C. Valdez
|
r865 | for n, ax in enumerate(self.axes): | ||
|
Juan C. Valdez
|
r866 | x, y, z = self.fill_gaps(*self.decimate()) | ||
|
José Chávez
|
r1004 | if self.xmin is None: | ||
| xmin = self.min_time | ||||
| else: | ||||
| xmin = fromtimestamp(int(self.xmin), self.min_time) | ||||
| if self.xmax is None: | ||||
| xmax = xmin + self.xrange*60*60 | ||||
| else: | ||||
| xmax = xmin + (self.xmax - self.xmin) * 60 * 60 | ||||
|
José Chávez
|
r931 | self.zmin = self.zmin if self.zmin else np.min(self.z) | ||
| self.zmax = self.zmax if self.zmax else np.max(self.z) | ||||
|
Juan C. Valdez
|
r865 | if ax.firsttime: | ||
| r889 | self.ymin = self.ymin if self.ymin else np.nanmin(self.y) | |||
| self.ymax = self.ymax if self.ymax else np.nanmax(self.y) | ||||
|
Juan C. Valdez
|
r866 | plot = ax.pcolormesh(x, y, z[n].T, | ||
| r889 | vmin=self.zmin, | |||
| vmax=self.zmax, | ||||
|
Juan C. Valdez
|
r865 | cmap=plt.get_cmap(self.colormap) | ||
| ) | ||||
| divider = make_axes_locatable(ax) | ||||
| r889 | cax = divider.new_horizontal(size='2%', pad=0.05) | |||
|
Juan C. Valdez
|
r865 | self.figure.add_axes(cax) | ||
| plt.colorbar(plot, cax) | ||||
|
Juan C. Valdez
|
r866 | ax.set_ylim(self.ymin, self.ymax) | ||
| r889 | ||||
| r922 | ax.xaxis.set_major_formatter(FuncFormatter(func)) | |||
| ax.xaxis.set_major_locator(LinearLocator(6)) | ||||
| r889 | ||||
| ax.set_ylabel(self.ylabel) | ||||
|
José Chávez
|
r898 | # if self.xmin is None: | ||
| # xmin = self.min_time | ||||
| # else: | ||||
| # xmin = (datetime.datetime.combine(self.dataOut.datatime.date(), | ||||
| # datetime.time(self.xmin, 0, 0))-d1970).total_seconds() | ||||
| r889 | ||||
| ax.set_xlim(xmin, xmax) | ||||
|
Juan C. Valdez
|
r865 | ax.firsttime = False | ||
| r889 | else: | |||
| ax.collections.remove(ax.collections[0]) | ||||
|
José Chávez
|
r898 | ax.set_xlim(xmin, xmax) | ||
| r889 | plot = ax.pcolormesh(x, y, z[n].T, | |||
| vmin=self.zmin, | ||||
| vmax=self.zmax, | ||||
| cmap=plt.get_cmap(self.colormap) | ||||
| ) | ||||
| r922 | ax.set_title('{} {}'.format(self.titles[n], | |||
| datetime.datetime.fromtimestamp(self.max_time).strftime('%y/%m/%d %H:%M:%S')), | ||||
| size=8) | ||||
| self.saveTime = self.min_time | ||||
| r889 | ||||
| class PlotCOHData(PlotRTIData): | ||||
| CODE = 'coh' | ||||
| def setup(self): | ||||
| self.ncols = 1 | ||||
| self.nrows = self.dataOut.nPairs | ||||
| self.width = 10 | ||||
| r922 | self.height = 2.2*self.nrows if self.nrows<6 else 12 | |||
| r907 | if self.nrows==1: | |||
| self.height += 1 | ||||
| r889 | self.ylabel = 'Range [Km]' | |||
| r922 | self.titles = ['{} Ch{} * Ch{}'.format(self.CODE.upper(), x[0], x[1]) for x in self.dataOut.pairsList] | |||
|
Juan C. Valdez
|
r865 | |||
| r889 | if self.figure is None: | |||
| self.figure = plt.figure(figsize=(self.width, self.height), | ||||
| edgecolor='k', | ||||
| facecolor='w') | ||||
| else: | ||||
| self.figure.clf() | ||||
| r907 | self.axes = [] | |||
| r889 | ||||
| for n in range(self.nrows): | ||||
| ax = self.figure.add_subplot(self.nrows, self.ncols, n+1) | ||||
| ax.firsttime = True | ||||
| self.axes.append(ax) | ||||
|
Juan C. Valdez
|
r865 | |||
| r907 | class PlotNoiseData(PlotData): | |||
| CODE = 'noise' | ||||
| def setup(self): | ||||
| self.ncols = 1 | ||||
| self.nrows = 1 | ||||
| self.width = 10 | ||||
| self.height = 3.2 | ||||
| self.ylabel = 'Intensity [dB]' | ||||
| self.titles = ['Noise'] | ||||
| r889 | ||||
| r907 | if self.figure is None: | |||
| self.figure = plt.figure(figsize=(self.width, self.height), | ||||
| edgecolor='k', | ||||
| facecolor='w') | ||||
| else: | ||||
| self.figure.clf() | ||||
| self.axes = [] | ||||
| self.ax = self.figure.add_subplot(self.nrows, self.ncols, 1) | ||||
| self.ax.firsttime = True | ||||
| def plot(self): | ||||
| x = self.times | ||||
| xmin = self.min_time | ||||
| xmax = xmin+self.xrange*60*60 | ||||
| if self.ax.firsttime: | ||||
| for ch in self.dataOut.channelList: | ||||
| y = [self.data[self.CODE][t][ch] for t in self.times] | ||||
| self.ax.plot(x, y, lw=1, label='Ch{}'.format(ch)) | ||||
| self.ax.firsttime = False | ||||
| self.ax.xaxis.set_major_formatter(FuncFormatter(func)) | ||||
| self.ax.xaxis.set_major_locator(LinearLocator(6)) | ||||
| self.ax.set_ylabel(self.ylabel) | ||||
| plt.legend() | ||||
| else: | ||||
| for ch in self.dataOut.channelList: | ||||
| y = [self.data[self.CODE][t][ch] for t in self.times] | ||||
| self.ax.lines[ch].set_data(x, y) | ||||
| self.ax.set_xlim(xmin, xmax) | ||||
| self.ax.set_ylim(min(y)-5, max(y)+5) | ||||
| r922 | self.saveTime = self.min_time | |||
| r907 | ||||
| r923 | class PlotWindProfilerData(PlotRTIData): | |||
|
Juan C. Espinoza
|
r937 | |||
| r923 | CODE = 'wind' | |||
| colormap = 'seismic' | ||||
| def setup(self): | ||||
| self.ncols = 1 | ||||
| self.nrows = self.dataOut.data_output.shape[0] | ||||
| self.width = 10 | ||||
| self.height = 2.2*self.nrows | ||||
| self.ylabel = 'Height [Km]' | ||||
|
Juan C. Espinoza
|
r937 | self.titles = ['Zonal Wind' ,'Meridional Wind', 'Vertical Wind'] | ||
| r923 | self.clabels = ['Velocity (m/s)','Velocity (m/s)','Velocity (cm/s)'] | |||
| self.windFactor = [1, 1, 100] | ||||
| if self.figure is None: | ||||
| self.figure = plt.figure(figsize=(self.width, self.height), | ||||
| edgecolor='k', | ||||
| facecolor='w') | ||||
| else: | ||||
| self.figure.clf() | ||||
| self.axes = [] | ||||
| for n in range(self.nrows): | ||||
| ax = self.figure.add_subplot(self.nrows, self.ncols, n+1) | ||||
| ax.firsttime = True | ||||
| self.axes.append(ax) | ||||
| def plot(self): | ||||
| self.x = np.array(self.times) | ||||
| self.y = self.dataOut.heightList | ||||
| self.z = [] | ||||
| for ch in range(self.nrows): | ||||
|
Juan C. Espinoza
|
r937 | self.z.append([self.data['output'][t][ch] for t in self.times]) | ||
| r923 | ||||
| self.z = np.array(self.z) | ||||
| self.z = numpy.ma.masked_invalid(self.z) | ||||
|
José Chávez
|
r927 | |||
| r923 | cmap=plt.get_cmap(self.colormap) | |||
|
Juan C. Espinoza
|
r937 | cmap.set_bad('black', 1.) | ||
| r923 | ||||
| for n, ax in enumerate(self.axes): | ||||
| x, y, z = self.fill_gaps(*self.decimate()) | ||||
| xmin = self.min_time | ||||
| xmax = xmin+self.xrange*60*60 | ||||
| if ax.firsttime: | ||||
| self.ymin = self.ymin if self.ymin else np.nanmin(self.y) | ||||
| self.ymax = self.ymax if self.ymax else np.nanmax(self.y) | ||||
| self.zmax = self.zmax if self.zmax else numpy.nanmax(abs(self.z[:-1, :])) | ||||
| self.zmin = self.zmin if self.zmin else -self.zmax | ||||
| plot = ax.pcolormesh(x, y, z[n].T*self.windFactor[n], | ||||
| vmin=self.zmin, | ||||
| vmax=self.zmax, | ||||
| cmap=cmap | ||||
| ) | ||||
| divider = make_axes_locatable(ax) | ||||
| cax = divider.new_horizontal(size='2%', pad=0.05) | ||||
| self.figure.add_axes(cax) | ||||
|
Juan C. Espinoza
|
r937 | cb = plt.colorbar(plot, cax) | ||
| cb.set_label(self.clabels[n]) | ||||
| r923 | ax.set_ylim(self.ymin, self.ymax) | |||
| ax.xaxis.set_major_formatter(FuncFormatter(func)) | ||||
| ax.xaxis.set_major_locator(LinearLocator(6)) | ||||
| ax.set_ylabel(self.ylabel) | ||||
| ax.set_xlim(xmin, xmax) | ||||
| ax.firsttime = False | ||||
| else: | ||||
| ax.collections.remove(ax.collections[0]) | ||||
| ax.set_xlim(xmin, xmax) | ||||
| plot = ax.pcolormesh(x, y, z[n].T*self.windFactor[n], | ||||
| vmin=self.zmin, | ||||
| vmax=self.zmax, | ||||
| cmap=plt.get_cmap(self.colormap) | ||||
| ) | ||||
| ax.set_title('{} {}'.format(self.titles[n], | ||||
| datetime.datetime.fromtimestamp(self.max_time).strftime('%y/%m/%d %H:%M:%S')), | ||||
| size=8) | ||||
| self.saveTime = self.min_time | ||||
| r907 | class PlotSNRData(PlotRTIData): | |||
|
José Chávez
|
r898 | CODE = 'snr' | ||
| r922 | colormap = 'jet' | |||
| r889 | ||||
|
José Chávez
|
r898 | class PlotDOPData(PlotRTIData): | ||
| CODE = 'dop' | ||||
| colormap = 'jet' | ||||
| r889 | ||||
| r922 | ||||
| r889 | class PlotPHASEData(PlotCOHData): | |||
| CODE = 'phase' | ||||
| colormap = 'seismic' | ||||
|
Juan C. Espinoza
|
r937 | |||
| class PlotSkyMapData(PlotData): | ||||
| CODE = 'met' | ||||
| def setup(self): | ||||
| self.ncols = 1 | ||||
| self.nrows = 1 | ||||
| self.width = 7.2 | ||||
| self.height = 7.2 | ||||
| self.xlabel = 'Zonal Zenith Angle (deg)' | ||||
| self.ylabel = 'Meridional Zenith Angle (deg)' | ||||
| if self.figure is None: | ||||
| self.figure = plt.figure(figsize=(self.width, self.height), | ||||
| edgecolor='k', | ||||
| facecolor='w') | ||||
| else: | ||||
| self.figure.clf() | ||||
| self.ax = plt.subplot2grid((self.nrows, self.ncols), (0, 0), 1, 1, polar=True) | ||||
| self.ax.firsttime = True | ||||
| def plot(self): | ||||
| arrayParameters = np.concatenate([self.data['param'][t] for t in self.times]) | ||||
| error = arrayParameters[:,-1] | ||||
| indValid = numpy.where(error == 0)[0] | ||||
| finalMeteor = arrayParameters[indValid,:] | ||||
| finalAzimuth = finalMeteor[:,3] | ||||
| finalZenith = finalMeteor[:,4] | ||||
| x = finalAzimuth*numpy.pi/180 | ||||
| y = finalZenith | ||||
| if self.ax.firsttime: | ||||
| self.ax.plot = self.ax.plot(x, y, 'bo', markersize=5)[0] | ||||
| self.ax.set_ylim(0,90) | ||||
| self.ax.set_yticks(numpy.arange(0,90,20)) | ||||
| self.ax.set_xlabel(self.xlabel) | ||||
| self.ax.set_ylabel(self.ylabel) | ||||
| self.ax.yaxis.labelpad = 40 | ||||
| self.ax.firsttime = False | ||||
| else: | ||||
| self.ax.plot.set_data(x, y) | ||||
| dt1 = datetime.datetime.fromtimestamp(self.min_time).strftime('%y/%m/%d %H:%M:%S') | ||||
| dt2 = datetime.datetime.fromtimestamp(self.max_time).strftime('%y/%m/%d %H:%M:%S') | ||||
| title = 'Meteor Detection Sky Map\n %s - %s \n Number of events: %5.0f\n' % (dt1, | ||||
| dt2, | ||||
| len(x)) | ||||
| self.ax.set_title(title, size=8) | ||||
| self.saveTime = self.max_time | ||||
