jroplot_parameters.py
627 lines
| 22.8 KiB
| text/x-python
|
PythonLexer
Julio Valdez
|
r502 | import os | ||
| import datetime | ||||
| import numpy | ||||
ebocanegra
|
r1001 | |||
Juan C. Espinoza
|
r1285 | from schainpy.model.graphics.jroplot_base import Plot, plt | ||
Danny Scipión
|
r1358 | from schainpy.model.graphics.jroplot_spectra import SpectraPlot, RTIPlot, CoherencePlot, SpectraCutPlot | ||
|
Juan C. Espinoza
|
r1285 | from schainpy.utils import log | ||
| r1367 | # libreria wradlib | |||
| import wradlib as wrl | ||||
|
ebocanegra
|
r1001 | |||
|
Juan C. Espinoza
|
r1285 | EARTH_RADIUS = 6.3710e3 | ||
|
ebocanegra
|
r1001 | |||
|
Juan C. Espinoza
|
r1285 | def ll2xy(lat1, lon1, lat2, lon2): | ||
|
ebocanegra
|
r1001 | |||
|
Juan C. Espinoza
|
r1285 | 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) | ||||
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | def km2deg(km): | ||
| ''' | ||||
| Convert distance in km to degrees | ||||
| ''' | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | return numpy.rad2deg(km/EARTH_RADIUS) | ||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | class SpectralMomentsPlot(SpectraPlot): | ||
| ''' | ||||
| Plot for Spectral Moments | ||||
| ''' | ||||
| CODE = 'spc_moments' | ||||
|
Danny Scipión
|
r1358 | # colormap = 'jet' | ||
| # plot_type = 'pcolor' | ||||
| class DobleGaussianPlot(SpectraPlot): | ||||
| ''' | ||||
| Plot for Double Gaussian Plot | ||||
| ''' | ||||
| CODE = 'gaussian_fit' | ||||
| # colormap = 'jet' | ||||
| # plot_type = 'pcolor' | ||||
|
José Chávez
|
r897 | |||
|
Danny Scipión
|
r1358 | class DoubleGaussianSpectraCutPlot(SpectraCutPlot): | ||
| ''' | ||||
| Plot SpectraCut with Double Gaussian Fit | ||||
| ''' | ||||
| CODE = 'cut_gaussian_fit' | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | class SnrPlot(RTIPlot): | ||
| ''' | ||||
| Plot for SNR Data | ||||
| ''' | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | CODE = 'snr' | ||
| colormap = 'jet' | ||||
|
José Chávez
|
r897 | |||
| r1343 | def update(self, dataOut): | |||
| data = { | ||||
| r1367 | 'snr': 10*numpy.log10(dataOut.data_snr) | |||
| r1343 | } | |||
| return data, {} | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | class DopplerPlot(RTIPlot): | ||
| ''' | ||||
| Plot for DOPPLER Data (1st moment) | ||||
| ''' | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | CODE = 'dop' | ||
| colormap = 'jet' | ||||
|
José Chávez
|
r897 | |||
| r1343 | def update(self, dataOut): | |||
| data = { | ||||
| r1367 | 'dop': 10*numpy.log10(dataOut.data_dop) | |||
| r1343 | } | |||
| return data, {} | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | class PowerPlot(RTIPlot): | ||
| ''' | ||||
| Plot for Power Data (0 moment) | ||||
| ''' | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | CODE = 'pow' | ||
| colormap = 'jet' | ||||
|
José Chávez
|
r897 | |||
| r1343 | def update(self, dataOut): | |||
| data = { | ||||
| r1367 | 'pow': 10*numpy.log10(dataOut.data_pow/dataOut.normFactor) | |||
| r1343 | } | |||
| return data, {} | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | class SpectralWidthPlot(RTIPlot): | ||
| ''' | ||||
| Plot for Spectral Width Data (2nd moment) | ||||
| ''' | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | CODE = 'width' | ||
| colormap = 'jet' | ||||
|
José Chávez
|
r897 | |||
| r1343 | def update(self, dataOut): | |||
| data = { | ||||
| 'width': dataOut.data_width | ||||
| } | ||||
| return data, {} | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | class SkyMapPlot(Plot): | ||
| ''' | ||||
| Plot for meteors detection data | ||||
| ''' | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | CODE = 'param' | ||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | def setup(self): | ||
|
Julio Valdez
|
r502 | |||
|
Juan C. Espinoza
|
r1285 | self.ncols = 1 | ||
| 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 | ||||
| self.ymin = -180 | ||||
| self.ymax = 180 | ||||
| self.colorbar = False | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | def plot(self): | ||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | arrayParameters = numpy.concatenate(self.data['param']) | ||
| error = arrayParameters[:, -1] | ||||
| indValid = numpy.where(error == 0)[0] | ||||
| finalMeteor = arrayParameters[indValid, :] | ||||
| finalAzimuth = finalMeteor[:, 3] | ||||
| finalZenith = finalMeteor[:, 4] | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | x = finalAzimuth * numpy.pi / 180 | ||
| y = finalZenith | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | ax = self.axes[0] | ||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | if ax.firsttime: | ||
| ax.plot = ax.plot(x, y, 'bo', markersize=5)[0] | ||||
|
Julio Valdez
|
r832 | else: | ||
|
Juan C. Espinoza
|
r1285 | ax.plot.set_data(x, y) | ||
| dt1 = self.getDateTime(self.data.min_time).strftime('%y/%m/%d %H:%M:%S') | ||||
| dt2 = self.getDateTime(self.data.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.titles[0] = title | ||||
| r1343 | class GenericRTIPlot(Plot): | |||
|
Juan C. Espinoza
|
r1285 | ''' | ||
| r1343 | Plot for data_xxxx object | |||
|
Juan C. Espinoza
|
r1285 | ''' | ||
| CODE = 'param' | ||||
| r1343 | colormap = 'viridis' | |||
| plot_type = 'pcolorbuffer' | ||||
|
Juan C. Espinoza
|
r1285 | |||
| def setup(self): | ||||
| self.xaxis = 'time' | ||||
| self.ncols = 1 | ||||
| r1359 | self.nrows = self.data.shape('param')[0] | |||
|
Juan C. Espinoza
|
r1285 | self.nplots = self.nrows | ||
|
Juan C. Espinoza
|
r1322 | self.plots_adjust.update({'hspace':0.8, 'left': 0.1, 'bottom': 0.08, 'right':0.95, 'top': 0.95}) | ||
| r1367 | ||||
|
Juan C. Espinoza
|
r1322 | if not self.xlabel: | ||
| self.xlabel = 'Time' | ||||
|
Juan C. Espinoza
|
r1285 | |||
|
Danny Scipión
|
r1358 | self.ylabel = 'Range [km]' | ||
|
Juan C. Espinoza
|
r1285 | if not self.titles: | ||
| r1360 | self.titles = ['Param {}'.format(x) for x in range(self.nrows)] | |||
|
Juan C. Espinoza
|
r1285 | |||
| r1343 | def update(self, dataOut): | |||
| data = { | ||||
| r1359 | 'param' : numpy.concatenate([getattr(dataOut, attr) for attr in self.attr_data], axis=0) | |||
| r1343 | } | |||
| meta = {} | ||||
| return data, meta | ||||
| r1367 | ||||
|
Juan C. Espinoza
|
r1285 | def plot(self): | ||
| r1343 | # self.data.normalize_heights() | |||
|
Juan C. Espinoza
|
r1285 | self.x = self.data.times | ||
| r1343 | self.y = self.data.yrange | |||
| r1359 | self.z = self.data['param'] | |||
|
José Chávez
|
r897 | |||
david c
|
r1378 | self.z = 10*numpy.log10(self.z) | ||
|
Juan C. Espinoza
|
r1285 | self.z = numpy.ma.masked_invalid(self.z) | ||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | if self.decimation is None: | ||
| x, y, z = self.fill_gaps(self.x, self.y, self.z) | ||||
|
Julio Valdez
|
r511 | else: | ||
|
Juan C. Espinoza
|
r1285 | x, y, z = self.fill_gaps(*self.decimate()) | ||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | for n, ax in enumerate(self.axes): | ||
Miguel Valdez
|
r588 | |||
|
Juan C. Espinoza
|
r1285 | self.zmax = self.zmax if self.zmax is not None else numpy.max( | ||
| self.z[n]) | ||||
| self.zmin = self.zmin if self.zmin is not None else numpy.min( | ||||
| self.z[n]) | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | if ax.firsttime: | ||
| if self.zlimits is not None: | ||||
| self.zmin, self.zmax = self.zlimits[n] | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | ax.plt = ax.pcolormesh(x, y, z[n].T * self.factors[n], | ||
| vmin=self.zmin, | ||||
| vmax=self.zmax, | ||||
| cmap=self.cmaps[n] | ||||
| ) | ||||
| else: | ||||
| if self.zlimits is not None: | ||||
| self.zmin, self.zmax = self.zlimits[n] | ||||
| ax.collections.remove(ax.collections[0]) | ||||
| ax.plt = ax.pcolormesh(x, y, z[n].T * self.factors[n], | ||||
| vmin=self.zmin, | ||||
| vmax=self.zmax, | ||||
| cmap=self.cmaps[n] | ||||
| ) | ||||
| class PolarMapPlot(Plot): | ||||
| ''' | ||||
| Plot for weather radar | ||||
| ''' | ||||
| CODE = 'param' | ||||
| colormap = 'seismic' | ||||
| def setup(self): | ||||
| self.ncols = 1 | ||||
| self.nrows = 1 | ||||
| self.width = 9 | ||||
| self.height = 8 | ||||
| self.mode = self.data.meta['mode'] | ||||
| if self.channels is not None: | ||||
| self.nplots = len(self.channels) | ||||
| self.nrows = len(self.channels) | ||||
| r1065 | else: | |||
|
Juan C. Espinoza
|
r1285 | self.nplots = self.data.shape(self.CODE)[0] | ||
| self.nrows = self.nplots | ||||
| self.channels = list(range(self.nplots)) | ||||
| if self.mode == 'E': | ||||
| self.xlabel = 'Longitude' | ||||
| self.ylabel = 'Latitude' | ||||
| else: | ||||
| self.xlabel = 'Range (km)' | ||||
| self.ylabel = 'Height (km)' | ||||
| self.bgcolor = 'white' | ||||
| self.cb_labels = self.data.meta['units'] | ||||
| self.lat = self.data.meta['latitude'] | ||||
| self.lon = self.data.meta['longitude'] | ||||
| self.xmin, self.xmax = float( | ||||
| km2deg(self.xmin) + self.lon), float(km2deg(self.xmax) + self.lon) | ||||
| self.ymin, self.ymax = float( | ||||
| km2deg(self.ymin) + self.lat), float(km2deg(self.ymax) + self.lat) | ||||
| # self.polar = True | ||||
| def plot(self): | ||||
| for n, ax in enumerate(self.axes): | ||||
| data = self.data['param'][self.channels[n]] | ||||
| zeniths = numpy.linspace( | ||||
| 0, self.data.meta['max_range'], data.shape[1]) | ||||
| if self.mode == 'E': | ||||
| r1343 | azimuths = -numpy.radians(self.data.yrange)+numpy.pi/2 | |||
|
Juan C. Espinoza
|
r1285 | r, theta = numpy.meshgrid(zeniths, azimuths) | ||
| x, y = r*numpy.cos(theta)*numpy.cos(numpy.radians(self.data.meta['elevation'])), r*numpy.sin( | ||||
| theta)*numpy.cos(numpy.radians(self.data.meta['elevation'])) | ||||
| x = km2deg(x) + self.lon | ||||
| y = km2deg(y) + self.lat | ||||
| r1065 | else: | |||
| r1343 | azimuths = numpy.radians(self.data.yrange) | |||
|
Juan C. Espinoza
|
r1285 | r, theta = numpy.meshgrid(zeniths, azimuths) | ||
| x, y = r*numpy.cos(theta), r*numpy.sin(theta) | ||||
| self.y = zeniths | ||||
| if ax.firsttime: | ||||
| if self.zlimits is not None: | ||||
| self.zmin, self.zmax = self.zlimits[n] | ||||
| ax.plt = ax.pcolormesh( # r, theta, numpy.ma.array(data, mask=numpy.isnan(data)), | ||||
| x, y, numpy.ma.array(data, mask=numpy.isnan(data)), | ||||
| vmin=self.zmin, | ||||
| vmax=self.zmax, | ||||
| cmap=self.cmaps[n]) | ||||
|
Julio Valdez
|
r513 | else: | ||
|
Juan C. Espinoza
|
r1285 | if self.zlimits is not None: | ||
| self.zmin, self.zmax = self.zlimits[n] | ||||
| ax.collections.remove(ax.collections[0]) | ||||
| ax.plt = ax.pcolormesh( # r, theta, numpy.ma.array(data, mask=numpy.isnan(data)), | ||||
| x, y, numpy.ma.array(data, mask=numpy.isnan(data)), | ||||
| vmin=self.zmin, | ||||
| vmax=self.zmax, | ||||
| cmap=self.cmaps[n]) | ||||
| if self.mode == 'A': | ||||
| continue | ||||
| # plot district names | ||||
| f = open('/data/workspace/schain_scripts/distrito.csv') | ||||
| for line in f: | ||||
| label, lon, lat = [s.strip() for s in line.split(',') if s] | ||||
| lat = float(lat) | ||||
| lon = float(lon) | ||||
| # ax.plot(lon, lat, '.b', ms=2) | ||||
| ax.text(lon, lat, label.decode('utf8'), ha='center', | ||||
| va='bottom', size='8', color='black') | ||||
| # plot limites | ||||
| limites = [] | ||||
| tmp = [] | ||||
| for line in open('/data/workspace/schain_scripts/lima.csv'): | ||||
| if '#' in line: | ||||
| if tmp: | ||||
| limites.append(tmp) | ||||
| tmp = [] | ||||
| continue | ||||
| values = line.strip().split(',') | ||||
| tmp.append((float(values[0]), float(values[1]))) | ||||
| for points in limites: | ||||
| ax.add_patch( | ||||
| Polygon(points, ec='k', fc='none', ls='--', lw=0.5)) | ||||
| # plot Cuencas | ||||
| for cuenca in ('rimac', 'lurin', 'mala', 'chillon', 'chilca', 'chancay-huaral'): | ||||
| f = open('/data/workspace/schain_scripts/{}.csv'.format(cuenca)) | ||||
| values = [line.strip().split(',') for line in f] | ||||
| points = [(float(s[0]), float(s[1])) for s in values] | ||||
| ax.add_patch(Polygon(points, ec='b', fc='none')) | ||||
| # plot grid | ||||
| for r in (15, 30, 45, 60): | ||||
| ax.add_artist(plt.Circle((self.lon, self.lat), | ||||
| km2deg(r), color='0.6', fill=False, lw=0.2)) | ||||
| ax.text( | ||||
| self.lon + (km2deg(r))*numpy.cos(60*numpy.pi/180), | ||||
| self.lat + (km2deg(r))*numpy.sin(60*numpy.pi/180), | ||||
| '{}km'.format(r), | ||||
| ha='center', va='bottom', size='8', color='0.6', weight='heavy') | ||||
| if self.mode == 'E': | ||||
| title = 'El={}$^\circ$'.format(self.data.meta['elevation']) | ||||
| label = 'E{:02d}'.format(int(self.data.meta['elevation'])) | ||||
|
Julio Valdez
|
r608 | else: | ||
|
Juan C. Espinoza
|
r1285 | title = 'Az={}$^\circ$'.format(self.data.meta['azimuth']) | ||
| label = 'A{:02d}'.format(int(self.data.meta['azimuth'])) | ||||
|
José Chávez
|
r897 | |||
|
Juan C. Espinoza
|
r1285 | self.save_labels = ['{}-{}'.format(lbl, label) for lbl in self.labels] | ||
| self.titles = ['{} {}'.format( | ||||
| self.data.parameters[x], title) for x in self.channels] | ||||
|
José Chávez
|
r897 | |||
| r1367 | class WeatherPlot(Plot): | |||
| CODE = 'weather' | ||||
| plot_name = 'weather' | ||||
| plot_type = 'ppistyle' | ||||
| buffering = False | ||||
| def setup(self): | ||||
| self.ncols = 1 | ||||
| self.nrows = 1 | ||||
| self.nplots= 1 | ||||
| self.ylabel= 'Range [Km]' | ||||
| self.titles= ['Weather'] | ||||
| self.colorbar=False | ||||
| self.width =8 | ||||
| self.height =8 | ||||
| self.ini =0 | ||||
| self.len_azi =0 | ||||
| self.buffer_ini = None | ||||
| self.buffer_azi = None | ||||
| self.plots_adjust.update({'wspace': 0.4, 'hspace':0.4, 'left': 0.1, 'right': 0.9, 'bottom': 0.08}) | ||||
| self.flag =0 | ||||
| self.indicador= 0 | ||||
| def update(self, dataOut): | ||||
| data = {} | ||||
| meta = {} | ||||
|
david c
|
r1378 | data['weather'] = 10*numpy.log10(dataOut.data_360[0]/(250**2)) | ||
| r1367 | data['azi'] = dataOut.data_azi | |||
| return data, meta | ||||
| def plot(self): | ||||
| thisDatetime = datetime.datetime.utcfromtimestamp(self.data.times[-1]) | ||||
|
david c
|
r1378 | print("--------------------------------------",self.ini,"-----------------------------------") | ||
| print("time",self.data.times[-1]) | ||||
| r1367 | data = self.data[-1] | |||
|
david c
|
r1378 | #print("debug_0", data) | ||
| r1367 | tmp_h = (data['weather'].shape[1])/10.0 | |||
|
david c
|
r1378 | #print("debug_1",tmp_h) | ||
| r1367 | stoprange = float(tmp_h*1.5)#stoprange = float(33*1.5) por ahora 400 | |||
| rangestep = float(0.15) | ||||
| r = numpy.arange(0, stoprange, rangestep) | ||||
| self.y = 2*r | ||||
|
david c
|
r1378 | print("---------------") | ||
| r1367 | tmp_v = data['weather'] | |||
|
david c
|
r1378 | #print("tmp_v",tmp_v.shape) | ||
| r1367 | tmp_z = data['azi'] | |||
|
david c
|
r1378 | print("tmp_z-------------->",tmp_z) | ||
| ##if self.ini==0: | ||||
| ## tmp_z= [0,1,2,3,4,5,6,7,8,9] | ||||
| r1367 | #print("tmp_z",tmp_z.shape) | |||
| res = 1 | ||||
| step = (360/(res*tmp_v.shape[0])) | ||||
|
david c
|
r1378 | #print("step",step) | ||
| r1367 | mode = 1 | |||
| if mode==0: | ||||
| #print("self.ini",self.ini) | ||||
| val = numpy.mean(tmp_v[:,0]) | ||||
| self.len_azi = len(tmp_z) | ||||
| ones = numpy.ones([(360-tmp_v.shape[0]),tmp_v.shape[1]])*val | ||||
| self.buffer_ini = numpy.vstack((tmp_v,ones)) | ||||
| n = ((360/res)-len(tmp_z)) | ||||
| start = tmp_z[-1]+res | ||||
| end = tmp_z[0]-res | ||||
| if start>end: | ||||
| end = end+360 | ||||
| azi_zeros = numpy.linspace(start,end,int(n)) | ||||
| azi_zeros = numpy.where(azi_zeros>360,azi_zeros-360,azi_zeros) | ||||
| self.buffer_ini_azi = numpy.hstack((tmp_z,azi_zeros)) | ||||
| self.ini = self.ini+1 | ||||
| if mode==1: | ||||
|
david c
|
r1378 | #print("################") | ||
| #print("################") | ||||
| #print("mode",self.ini) | ||||
| r1367 | #print("self.ini",self.ini) | |||
| if self.ini==0: | ||||
| res = 1 | ||||
| step = (360/(res*tmp_v.shape[0])) | ||||
| val = numpy.mean(tmp_v[:,0]) | ||||
| self.len_azi = len(tmp_z) | ||||
| self.buf_tmp = tmp_v | ||||
| ones = numpy.ones([(360-tmp_v.shape[0]),tmp_v.shape[1]])*val | ||||
| self.buffer_ini = numpy.vstack((tmp_v,ones)) | ||||
| n = ((360/res)-len(tmp_z)) | ||||
| start = tmp_z[-1]+res | ||||
| end = tmp_z[0]-res | ||||
| if start>end: | ||||
| end =end+360 | ||||
| azi_zeros = numpy.linspace(start,end,int(n)) | ||||
| azi_zeros = numpy.where(azi_zeros>360,azi_zeros-360,azi_zeros) | ||||
| self.buf_azi = tmp_z | ||||
| self.buffer_ini_azi = numpy.hstack((tmp_z,azi_zeros)) | ||||
| self.ini = self.ini+1 | ||||
| elif 0<self.ini<step: | ||||
| ''' | ||||
| if self.ini>31: | ||||
| start= tmp_z[0] | ||||
| end =tmp_z[-1] | ||||
| print("start","end",start,end) | ||||
| if self.ini==32: | ||||
| tmp_v=tmp_v+20 | ||||
| if self.ini==33: | ||||
| tmp_v=tmp_v+10 | ||||
| if self.ini==34: | ||||
| tmp_v=tmp_v+20 | ||||
| if self.ini==35: | ||||
| tmp_v=tmp_v+20 | ||||
| ''' | ||||
| self.buf_tmp= numpy.vstack((self.buf_tmp,tmp_v)) | ||||
| print("ERROR_INMINENTE",self.buf_tmp.shape) | ||||
| if self.buf_tmp.shape[0]==360: | ||||
|
david c
|
r1378 | print("entre aqui en 360 grados") | ||
| r1367 | self.buffer_ini=self.buf_tmp | |||
| else: | ||||
|
david c
|
r1378 | # nuevo######### | ||
| self.buffer_ini[0:self.buf_tmp.shape[0],:]=self.buf_tmp | ||||
| ################ | ||||
| #val=30.0 | ||||
| #ones = numpy.ones([(360-self.buf_tmp.shape[0]),self.buf_tmp.shape[1]])*val | ||||
| #self.buffer_ini = numpy.vstack((self.buf_tmp,ones)) | ||||
| r1367 | ||||
| self.buf_azi = numpy.hstack((self.buf_azi,tmp_z)) | ||||
| n = ((360/res)-len(self.buf_azi)) | ||||
|
david c
|
r1378 | print("n----->",n) | ||
| r1367 | if n==0: | |||
| self.buffer_ini_azi = self.buf_azi | ||||
| else: | ||||
| start = self.buf_azi[-1]+res | ||||
| end = self.buf_azi[0]-res | ||||
|
david c
|
r1378 | print("start",start) | ||
| print("end",end) | ||||
| r1367 | if start>end: | |||
| end =end+360 | ||||
| azi_zeros = numpy.linspace(start,end,int(n)) | ||||
| azi_zeros = numpy.where(azi_zeros>360,azi_zeros-360,azi_zeros) | ||||
|
david c
|
r1378 | print("self.buf_azi",self.buf_azi[0]) | ||
| print("tmp_Z 0 ",tmp_z[0]) | ||||
| print("tmp_Z -1",tmp_z[-1]) | ||||
| r1367 | if tmp_z[0]<self.buf_azi[0] <tmp_z[-1]: | |||
|
david c
|
r1378 | print("activando indicador") | ||
| r1367 | self.indicador=1 | |||
| if self.indicador==1: | ||||
| azi_zeros = numpy.ones(360-len(self.buf_azi))*(tmp_z[-1]+res) | ||||
|
david c
|
r1378 | ###start = tmp_z[-1]+res | ||
| ###end = tmp_z[0]-res | ||||
| ###if start>end: | ||||
| ### end =end+360 | ||||
| ###azi_zeros = numpy.linspace(start,end,int(n)) | ||||
| ###azi_zeros = numpy.where(azi_zeros>360,azi_zeros-360,azi_zeros) | ||||
| print("azi_zeros",azi_zeros) | ||||
| ######self.buffer_ini_azi = numpy.hstack((self.buf_azi,azi_zeros)) | ||||
| #self.buffer_ini[0:tmv.shape[0],:]=tmp_v | ||||
| ##self.indicador=0 | ||||
| r1367 | # self.indicador = True | |||
| #if self.indicador==True: | ||||
| # azi_zeros = numpy.ones(360-len(self.buf_azi))*(tmp_z[-1]+res) | ||||
| #self.buf_azi = tmp_z | ||||
| self.buffer_ini_azi = numpy.hstack((self.buf_azi,azi_zeros)) | ||||
| if self.ini==step-1: | ||||
| start= tmp_z[0] | ||||
| end = tmp_z[-1] | ||||
| #print("start","end",start,end) | ||||
| ###print(self.buffer_ini_azi[:80]) | ||||
| self.ini = self.ini+1 | ||||
| else: | ||||
| step = (360/(res*tmp_v.shape[0])) | ||||
|
david c
|
r1378 | # aqui estaba realizando el debug de simulacion | ||
| # tmp_v=tmp_v +5 en cada step sumaba 5 | ||||
| # y el mismo valor despues de la primera vuelta | ||||
| #tmp_v=tmp_v+5+(self.ini-step)*1### aqui yo habia sumado 5 por las puras | ||||
| r1367 | ||||
| start= tmp_z[0] | ||||
| end = tmp_z[-1] | ||||
| #print("start","end",start,end) | ||||
| ###print(self.buffer_ini_azi[:120]) | ||||
| if step>=2: | ||||
| if self.flag<step-1: | ||||
| limit_i=self.buf_azi[len(tmp_z)*(self.flag+1)] | ||||
| limit_s=self.buf_azi[len(tmp_z)*(self.flag+2)-1] | ||||
| print("flag",self.flag,limit_i,limit_s) | ||||
| if limit_i< tmp_z[-1]< limit_s: | ||||
| index_i=int(numpy.where(tmp_z<=self.buf_azi[len(tmp_z)*(self.flag+1)])[0][-1]) | ||||
| tmp_r =int(numpy.where(self.buf_azi[(self.flag+1)*len(tmp_z):(self.flag+2)*len(tmp_z)]>=tmp_z[-1])[0][0]) | ||||
| print("tmp_r",tmp_r) | ||||
| index_f=(self.flag+1)*len(tmp_z)+tmp_r | ||||
| if len(tmp_z[index_i:])>len(self.buf_azi[len(tmp_z)*(self.flag+1):index_f]): | ||||
| final = len(self.buf_azi[len(tmp_z)*(self.flag+1):index_f]) | ||||
| else: | ||||
| final= len(tmp_z[index_i:]) | ||||
| self.buf_azi[len(tmp_z)*(self.flag+1):index_f]=tmp_z[index_i:index_i+final] | ||||
| self.buf_tmp[len(tmp_z)*(self.flag+1):index_f,:]=tmp_v[index_i:index_i+final,:] | ||||
| if limit_i<tmp_z[0]<limit_s: | ||||
| index_f =int(numpy.where(self.buf_azi>=tmp_z[-1])[0][0]) | ||||
| n_p =index_f-len(tmp_z)*(self.flag+1) | ||||
| if n_p>0: | ||||
| self.buf_azi[len(tmp_z)*(self.flag+1):index_f]=tmp_z[-1]*numpy.ones(n_p) | ||||
| self.buf_tmp[len(tmp_z)*(self.flag+1):index_f,:]=tmp_v[-1,:]*numpy.ones([n_p,tmp_v.shape[1]]) | ||||
| ''' | ||||
| if self.buf_azi[len(tmp_z)]<tmp_z[-1]<self.buf_azi[2*len(tmp_z)-1]: | ||||
| index_i= int(numpy.where(tmp_z <= self.buf_azi[len(tmp_z)])[0][-1]) | ||||
| index_f= int(numpy.where(self.buf_azi>=tmp_z[-1])[0][0]) | ||||
| #print("index",index_i,index_f) | ||||
| if len(tmp_z[index_i:])>len(self.buf_azi[len(tmp_z):index_f]): | ||||
| final = len(self.buf_azi[len(tmp_z):index_f]) | ||||
| else: | ||||
| final = len(tmp_z[index_i:]) | ||||
| self.buf_azi[len(tmp_z):index_f]=tmp_z[index_i:index_i+final] | ||||
| self.buf_tmp[len(tmp_z):index_f,:]=tmp_v[index_i:index_i+final,:] | ||||
| ''' | ||||
| self.buf_tmp[len(tmp_z)*(self.flag):len(tmp_z)*(self.flag+1),:]=tmp_v | ||||
| self.buf_azi[len(tmp_z)*(self.flag):len(tmp_z)*(self.flag+1)] = tmp_z | ||||
| self.buffer_ini=self.buf_tmp | ||||
| self.buffer_ini_azi = self.buf_azi | ||||
| ##print("--------salida------------") | ||||
| start= tmp_z[0] | ||||
| end = tmp_z[-1] | ||||
| ##print("start","end",start,end) | ||||
| ##print(self.buffer_ini_azi[:120]) | ||||
| self.ini= self.ini+1 | ||||
| self.flag = self.flag +1 | ||||
| if self.flag==step: | ||||
| self.flag=0 | ||||
|
david c
|
r1378 | numpy.set_printoptions(suppress=True) | ||
| print("buffer_ini_azi") | ||||
| print(self.buffer_ini_azi) | ||||
| r1367 | for i,ax in enumerate(self.axes): | |||
| if ax.firsttime: | ||||
| plt.clf() | ||||
|
david c
|
r1378 | cgax, pm = wrl.vis.plot_ppi(self.buffer_ini,r=r,az=self.buffer_ini_azi,fig=self.figures[0], proj='cg', vmin=1, vmax=60) | ||
| r1367 | else: | |||
| plt.clf() | ||||
|
david c
|
r1378 | cgax, pm = wrl.vis.plot_ppi(self.buffer_ini,r=r,az=self.buffer_ini_azi,fig=self.figures[0], proj='cg', vmin=1, vmax=60) | ||
| r1367 | caax = cgax.parasites[0] | |||
| paax = cgax.parasites[1] | ||||
| cbar = plt.gcf().colorbar(pm, pad=0.075) | ||||
| caax.set_xlabel('x_range [km]') | ||||
| caax.set_ylabel('y_range [km]') | ||||
|
david c
|
r1378 | plt.text(1.0, 1.05, 'azimuth '+str(thisDatetime)+"step"+str(self.ini), transform=caax.transAxes, va='bottom',ha='right') | ||
| #import time | ||||
| #time.sleep(0.5) | ||||
