schain-jc Commit - r1095:c7ed06c70307

Fix SkyMapPlot

Juan C. Espinoza -

r1095:c7ed06c70307

parent child

Context file:

r1095:c7ed06c70307

Collapse all files

schainpy/model/graphics/jroplot_data.py +41 -44

             from schainpy.model.proc.jroproc_base import Operation
             from schainpy.utils import log
-            jet_values = matplotlib.pyplot.get_cmap("jet", 100)(numpy.arange(100))[10:90]
+            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]
+                'seismic_r', 20)(numpy.arange(20))[10:15]
             ncmap = matplotlib.colors.LinearSegmentedColormap.from_list(
-                "jro", numpy.vstack((blu_values, jet_values)))
+                'jro', numpy.vstack((blu_values, jet_values)))
             matplotlib.pyplot.register_cmap(cmap=ncmap)
             CMAPS = [plt.get_cmap(s) for s in ('jro', 'jet', 'RdBu_r', 'seismic')]
                     self.colorbar = kwargs.get('colorbar', True)
                     self.factors = kwargs.get('factors', [1, 1, 1, 1, 1, 1, 1, 1])
                     self.titles = ['' for __ in range(16)]
+                    self.polar = False
                 def __fmtTime(self, x, pos):
                     '''
                     '''
                     Common setup for all figures, here figures and axes are created
                     '''
+                    if self.CODE not in self.data:
+                        raise ValueError(log.error('Missing data for {}'.format(self.CODE),
+                            self.name))
                     self.setup()
                                          facecolor='w')
                         self.figures.append(fig)
                         for n in range(self.nplots):
-                            ax = fig.add_subplot(self.nrows, self.ncols, n + 1)
+                            ax = fig.add_subplot(self.nrows, self.ncols, n + 1, polar=self.polar)
                             ax.tick_params(labelsize=8)
                             ax.firsttime = True
                             ax.index = 0
                             fig = plt.figure(figsize=(self.width, self.height),
                                              edgecolor='k',
                                              facecolor='w')
-                            ax = fig.add_subplot(1, 1, 1)
+                            ax = fig.add_subplot(1, 1, 1, polar=self.polar)
                             ax.tick_params(labelsize=8)
                             ax.firsttime = True
                             ax.index = 0
                     else:
                         if self.xaxis is 'time':
                             dt = self.getDateTime(self.max_time)
-                            xmax = (dt.replace(hour=int(self.xmax), minute=0, second=0) - datetime.datetime(1970, 1, 1)).total_seconds()
+                            xmax = (dt.replace(hour=int(self.xmax), minute=59, second=59) - datetime.datetime(1970, 1, 1)).total_seconds()
                             if self.data.localtime:
                                 xmax += time.timezone
                         else:
                     i = 1 if numpy.where(ymax < Y)[0][0] < 0 else numpy.where(ymax < Y)[0][0]
                     ystep = Y[i-1]/5
-                    ystep = 200 if ymax >= 800 else 100 if ymax >= 400 else 50 if ymax >= 200 else 20
                     for n, ax in enumerate(self.axes):
                         if ax.firsttime:
                             ax.set_facecolor(self.bgcolor)
                                 [tick.set_visible(False)
                                  for tick in self.pf_axes[n].get_yticklabels()]
                             if self.colorbar:
-                                ax.cbar = plt.colorbar(ax.plt, ax=ax, pad=0.02, aspect=10)
+                                ax.cbar = plt.colorbar(ax.plt, ax=ax, fraction=0.1, 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_labels[n], size=8)
                             else:
                                 ax.cbar = None
-                        ax.set_title('{} - {} {}'.format(
+                        if not self.polar:
+                            ax.set_xlim(xmin, xmax)
+                            ax.set_ylim(ymin, ymax)
+                            ax.set_title('{} - {} {}'.format(
                                 self.titles[n],
                                 self.getDateTime(self.max_time).strftime('%H:%M:%S'),
                                 self.time_label),
                             size=8)
-                        ax.set_xlim(xmin, xmax)
+                        else:
-                        ax.set_ylim(ymin, ymax)
+                            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
                 def __plot(self):
                     '''
                     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')
                             continue
                         fig.tight_layout()
                                     self.getDateTime(self.saveTime).strftime('%y%m%d_%H%M%S')
                                 )
                             )
-                            print 'Saving figure: {}'.format(figname)
+                            log.log('Saving figure: {}'.format(figname), self.name)
                             fig.savefig(figname)
                 def plot(self):
                 Plot for meteors detection data
                 '''
-                CODE = 'met'
+                CODE = 'param'
                 def setup(self):
                     self.nrows = 1
                     self.width = 7.2
                     self.height = 7.2
+                    self.nplots = 1
                     self.xlabel = 'Zonal Zenith Angle (deg)'
                     self.ylabel = 'Meridional Zenith Angle (deg)'
+                    self.polar = True
-                    if self.figure is None:
+                    self.ymin = -180
-                        self.figure = plt.figure(figsize=(self.width, self.height),
+                    self.ymax = 180
-                                                 edgecolor='k',
+                    self.colorbar = False
-                                                 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 = numpy.concatenate(
+                    arrayParameters = numpy.concatenate(self.data['param'])
-                        [self.data['param'][t] for t in self.times])
                     error = arrayParameters[:, -1]
                     indValid = numpy.where(error == 0)[0]
                     finalMeteor = arrayParameters[indValid, :]
                     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)
+                    ax = self.axes[0]
+                    if ax.firsttime:
+                        ax.plot = ax.plot(x, y, 'bo', markersize=5)[0]
+                    else:
+                        ax.plot.set_data(x, y)
                     dt1 = self.getDateTime(self.min_time).strftime('%y/%m/%d %H:%M:%S')
                     dt2 = self.getDateTime(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.titles[0] = title
                     self.saveTime = self.max_time
                         )
                     else:
                         self.z = self.data[self.CODE]
                     self.z = numpy.ma.masked_invalid(self.z)
                     for n, ax in enumerate(self.axes):

General Comments 0

Write
Preview

You need to be logged in to leave comments. Login now

No TODOs yet

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages