@@ -22,7 +22,7 ncmap = matplotlib.colors.LinearSegmentedColormap.from_list( | |||||
22 | 'jro', numpy.vstack((blu_values, jet_values))) |
|
22 | 'jro', numpy.vstack((blu_values, jet_values))) | |
23 | matplotlib.pyplot.register_cmap(cmap=ncmap) |
|
23 | matplotlib.pyplot.register_cmap(cmap=ncmap) | |
24 |
|
24 | |||
25 |
CMAPS = [plt.get_cmap(s) for s in ('jro', 'jet', ' |
|
25 | CMAPS = [plt.get_cmap(s) for s in ('jro', 'jet', 'viridis', 'plasma', 'inferno', 'Greys', 'seismic', 'bwr', 'coolwarm')] | |
26 |
|
26 | |||
27 |
|
27 | |||
28 | def figpause(interval): |
|
28 | def figpause(interval): | |
@@ -45,8 +45,7 class PlotData(Operation, Process): | |||||
45 | CODE = 'Figure' |
|
45 | CODE = 'Figure' | |
46 | colormap = 'jro' |
|
46 | colormap = 'jro' | |
47 | bgcolor = 'white' |
|
47 | bgcolor = 'white' | |
48 | CONFLATE = False |
|
48 | CONFLATE = False | |
49 | __MAXNUMX = 80 |
|
|||
50 | __missing = 1E30 |
|
49 | __missing = 1E30 | |
51 |
|
50 | |||
52 | __attrs__ = ['show', 'save', 'xmin', 'xmax', 'ymin', 'ymax', 'zmin', 'zmax', |
|
51 | __attrs__ = ['show', 'save', 'xmin', 'xmax', 'ymin', 'ymax', 'zmin', 'zmax', | |
@@ -88,7 +87,7 class PlotData(Operation, Process): | |||||
88 | self.ymin = kwargs.get('ymin', None) |
|
87 | self.ymin = kwargs.get('ymin', None) | |
89 | self.ymax = kwargs.get('ymax', None) |
|
88 | self.ymax = kwargs.get('ymax', None) | |
90 | self.xlabel = kwargs.get('xlabel', None) |
|
89 | self.xlabel = kwargs.get('xlabel', None) | |
91 |
self. |
|
90 | self.decimation = kwargs.get('decimation', None) | |
92 | self.showSNR = kwargs.get('showSNR', False) |
|
91 | self.showSNR = kwargs.get('showSNR', False) | |
93 | self.oneFigure = kwargs.get('oneFigure', True) |
|
92 | self.oneFigure = kwargs.get('oneFigure', True) | |
94 | self.width = kwargs.get('width', None) |
|
93 | self.width = kwargs.get('width', None) | |
@@ -328,7 +327,7 class PlotData(Operation, Process): | |||||
328 | def decimate(self): |
|
327 | def decimate(self): | |
329 |
|
328 | |||
330 | # dx = int(len(self.x)/self.__MAXNUMX) + 1 |
|
329 | # dx = int(len(self.x)/self.__MAXNUMX) + 1 | |
331 |
dy = int(len(self.y) / self. |
|
330 | dy = int(len(self.y) / self.decimation) + 1 | |
332 |
|
331 | |||
333 | # x = self.x[::dx] |
|
332 | # x = self.x[::dx] | |
334 | x = self.x |
|
333 | x = self.x | |
@@ -369,7 +368,7 class PlotData(Operation, Process): | |||||
369 | ymin = self.ymin if self.ymin else numpy.nanmin(self.y) |
|
368 | ymin = self.ymin if self.ymin else numpy.nanmin(self.y) | |
370 | ymax = self.ymax if self.ymax else numpy.nanmax(self.y) |
|
369 | ymax = self.ymax if self.ymax else numpy.nanmax(self.y) | |
371 |
|
370 | |||
372 | Y = numpy.array([10, 20, 50, 100, 200, 500, 1000, 2000]) |
|
371 | Y = numpy.array([5, 10, 20, 50, 100, 200, 500, 1000, 2000]) | |
373 | i = 1 if numpy.where(ymax-ymin < Y)[0][0] < 0 else numpy.where(ymax-ymin < Y)[0][0] |
|
372 | i = 1 if numpy.where(ymax-ymin < Y)[0][0] < 0 else numpy.where(ymax-ymin < Y)[0][0] | |
374 | ystep = Y[i] / 5 |
|
373 | ystep = Y[i] / 5 | |
375 |
|
374 | |||
@@ -712,8 +711,12 class PlotRTIData(PlotData): | |||||
712 | self.z = self.data[self.CODE] |
|
711 | self.z = self.data[self.CODE] | |
713 | self.z = numpy.ma.masked_invalid(self.z) |
|
712 | self.z = numpy.ma.masked_invalid(self.z) | |
714 |
|
713 | |||
715 | for n, ax in enumerate(self.axes): |
|
714 | if self.decimation is None: | |
|
715 | x, y, z = self.fill_gaps(self.x, self.y, self.z) | |||
|
716 | else: | |||
716 | x, y, z = self.fill_gaps(*self.decimate()) |
|
717 | x, y, z = self.fill_gaps(*self.decimate()) | |
|
718 | ||||
|
719 | for n, ax in enumerate(self.axes): | |||
717 | self.zmin = self.zmin if self.zmin else numpy.min(self.z) |
|
720 | self.zmin = self.zmin if self.zmin else numpy.min(self.z) | |
718 | self.zmax = self.zmax if self.zmax else numpy.max(self.z) |
|
721 | self.zmax = self.zmax if self.zmax else numpy.max(self.z) | |
719 | if ax.firsttime: |
|
722 | if ax.firsttime: | |
@@ -916,9 +919,13 class PlotParamData(PlotRTIData): | |||||
916 |
|
919 | |||
917 | self.z = numpy.ma.masked_invalid(self.z) |
|
920 | self.z = numpy.ma.masked_invalid(self.z) | |
918 |
|
921 | |||
919 | for n, ax in enumerate(self.axes): |
|
922 | if self.decimation is None: | |
920 |
|
923 | x, y, z = self.fill_gaps(self.x, self.y, self.z) | ||
|
924 | else: | |||
921 | x, y, z = self.fill_gaps(*self.decimate()) |
|
925 | x, y, z = self.fill_gaps(*self.decimate()) | |
|
926 | ||||
|
927 | for n, ax in enumerate(self.axes): | |||
|
928 | ||||
922 | self.zmax = self.zmax if self.zmax is not None else numpy.max( |
|
929 | self.zmax = self.zmax if self.zmax is not None else numpy.max( | |
923 | self.z[n]) |
|
930 | self.z[n]) | |
924 | self.zmin = self.zmin if self.zmin is not None else numpy.min( |
|
931 | self.zmin = self.zmin if self.zmin is not None else numpy.min( |
General Comments 0
You need to be logged in to leave comments.
Login now