| @@ -1,3 +1,4 | |||
|
|
1 | # MASTER | |
|
|
1 | 2 | # Copyright (c) 2012-2021 Jicamarca Radio Observatory |
|
|
2 | 3 | # All rights reserved. |
|
|
3 | 4 | # |
| @@ -43,6 +44,11 class SpectraPlot(Plot): | |||
|
|
43 | 44 | spc = 10*numpy.log10(dataOut.data_spc/dataOut.normFactor) |
|
|
44 | 45 | data['spc'] = spc |
|
|
45 | 46 | data['rti'] = dataOut.getPower() |
|
|
47 | if hasattr(dataOut, 'LagPlot'): #Double Pulse | |
|
|
48 | max_hei_id = dataOut.nHeights - 2*dataOut.LagPlot | |
|
|
49 | data['noise'] = 10*numpy.log10(dataOut.getNoise(ymin_index=53,ymax_index=max_hei_id)/dataOut.normFactor) | |
|
|
50 | data['noise'][0] = 10*numpy.log10(dataOut.getNoise(ymin_index=53)[0]/dataOut.normFactor) | |
|
|
51 | else: | |
|
|
46 | 52 | data['noise'] = 10*numpy.log10(dataOut.getNoise()/dataOut.normFactor) |
|
|
47 | 53 | meta['xrange'] = (dataOut.getFreqRange(1)/1000., dataOut.getAcfRange(1), dataOut.getVelRange(1)) |
|
|
48 | 54 | |
| @@ -66,6 +72,7 class SpectraPlot(Plot): | |||
|
|
66 | 72 | self.xlabel = "Velocity (m/s)" |
|
|
67 | 73 | |
|
|
68 | 74 | if (self.CODE == 'spc_moments') | (self.CODE == 'gaussian_fit'): |
|
|
75 | print("data",self.data.shape) | |
|
|
69 | 76 | x = self.data.xrange[2] |
|
|
70 | 77 | self.xlabel = "Velocity (m/s)" |
|
|
71 | 78 | |
| @@ -77,9 +84,10 class SpectraPlot(Plot): | |||
|
|
77 | 84 | data = self.data[-1] |
|
|
78 | 85 | z = data['spc'] |
|
|
79 | 86 | |
|
|
87 | self.CODE2 = 'spc_oblique' | |
|
|
88 | ||
|
|
80 | 89 | for n, ax in enumerate(self.axes): |
|
|
81 | 90 | noise = data['noise'][n] |
|
|
82 | ||
|
|
83 | 91 | if self.CODE == 'spc_moments': |
|
|
84 | 92 | mean = data['moments'][n, 1] |
|
|
85 | 93 | if self.CODE == 'gaussian_fit': |
| @@ -87,13 +95,16 class SpectraPlot(Plot): | |||
|
|
87 | 95 | gau1 = data['gaussfit'][n][2,:,1] |
|
|
88 | 96 | if ax.firsttime: |
|
|
89 | 97 | self.xmax = self.xmax if self.xmax else numpy.nanmax(x) |
|
|
90 |
self.xmin = self.xmin if self.xmin else |
|
|
|
91 | self.zmin = self.zmin if self.zmin else numpy.nanmin(z) | |
|
|
92 | self.zmax = self.zmax if self.zmax else numpy.nanmax(z) | |
|
|
98 | self.xmin = self.xmin if self.xmin else numpy.nanmin(x)#-self.xmax | |
|
|
99 | #self.zmin = self.zmin if self.zmin else numpy.nanmin(z) | |
|
|
100 | #self.zmax = self.zmax if self.zmax else numpy.nanmax(z) | |
|
|
101 | if self.zlimits is not None: | |
|
|
102 | self.zmin, self.zmax = self.zlimits[n] | |
|
|
103 | # CORREGIR CON matplotlib==3.3.4 | |
|
|
93 | 104 | ax.plt = ax.pcolormesh(x, y, z[n].T, |
|
|
94 | 105 | vmin=self.zmin, |
|
|
95 | 106 | vmax=self.zmax, |
|
|
96 | cmap=plt.get_cmap(self.colormap) | |
|
|
107 | cmap=plt.get_cmap(self.colormap), | |
|
|
97 | 108 | ) |
|
|
98 | 109 | |
|
|
99 | 110 | if self.showprofile: |
| @@ -107,6 +118,8 class SpectraPlot(Plot): | |||
|
|
107 | 118 | ax.plt_gau0 = ax.plot(gau0, y, color='r', lw=1)[0] |
|
|
108 | 119 | ax.plt_gau1 = ax.plot(gau1, y, color='y', lw=1)[0] |
|
|
109 | 120 | else: |
|
|
121 | if self.zlimits is not None: | |
|
|
122 | self.zmin, self.zmax = self.zlimits[n] | |
|
|
110 | 123 | ax.plt.set_array(z[n].T.ravel()) |
|
|
111 | 124 | if self.showprofile: |
|
|
112 | 125 | ax.plt_profile.set_data(data['rti'][n], y) |
| @@ -374,7 +387,6 class CrossSpectra4Plot(Plot): | |||
|
|
374 | 387 | cmap=plt.get_cmap(self.colormap) |
|
|
375 | 388 | ) |
|
|
376 | 389 | else: |
|
|
377 | ||
|
|
378 | 390 | ax.plt.set_array(nspc[pair[0]].T.ravel()) |
|
|
379 | 391 | self.titles.append('CH {}: {:3.2f}dB'.format(pair[0], noise[pair[0]])) |
|
|
380 | 392 | |
General Comments 0
You need to be logged in to leave comments.
Login now