| @@ -379,3 +379,234 class PolarMapPlot(Plot): | |||
|
|
379 | 379 | self.save_labels = ['{}-{}'.format(lbl, label) for lbl in self.labels] |
|
|
380 | 380 | self.titles = ['{} {}'.format( |
|
|
381 | 381 | self.data.parameters[x], title) for x in self.channels] |
|
|
382 | ||
|
|
383 | class MP150KmRTIPlot(Plot): | |
|
|
384 | ''' | |
|
|
385 | Plot for data_xxxx object | |
|
|
386 | ''' | |
|
|
387 | ||
|
|
388 | CODE = 'param' | |
|
|
389 | colormap = 'viridis' | |
|
|
390 | plot_type = 'pcolorbuffer' | |
|
|
391 | ||
|
|
392 | def setup(self): | |
|
|
393 | self.xaxis = 'time' | |
|
|
394 | self.ncols = 1 | |
|
|
395 | self.nrows = self.data.shape('param')[0] | |
|
|
396 | self.nplots = self.nrows | |
|
|
397 | self.plots_adjust.update({'hspace':0.8, 'left': 0.1, 'bottom': 0.08, 'right':0.95, 'top': 0.95}) | |
|
|
398 | ||
|
|
399 | if not self.xlabel: | |
|
|
400 | self.xlabel = 'Time' | |
|
|
401 | ||
|
|
402 | self.ylabel = 'Range [km]' | |
|
|
403 | if not self.titles: | |
|
|
404 | self.titles = ['Param {}'.format(x) for x in range(self.nrows)] | |
|
|
405 | ||
|
|
406 | def update(self, dataOut): | |
|
|
407 | data = { | |
|
|
408 | #'param' : numpy.concatenate([getattr(dataOut, attr) for attr in self.attr_data], axis=0)[0:3,:] # SNL, VERTICAL, ZONAL | |
|
|
409 | 'param' : dataOut.data_output[0:3,:] # SNL, VERTICAL, ZONAL | |
|
|
410 | } | |
|
|
411 | ||
|
|
412 | meta = {} | |
|
|
413 | ||
|
|
414 | return data, meta | |
|
|
415 | ||
|
|
416 | def plot(self): | |
|
|
417 | # self.data.normalize_heights() | |
|
|
418 | self.x = self.data.times | |
|
|
419 | self.y = self.data.yrange | |
|
|
420 | self.z = self.data['param'] | |
|
|
421 | ||
|
|
422 | ||
|
|
423 | self.z = numpy.ma.masked_invalid(self.z) | |
|
|
424 | ||
|
|
425 | if self.decimation is None: | |
|
|
426 | x, y, z = self.fill_gaps(self.x, self.y, self.z) | |
|
|
427 | else: | |
|
|
428 | x, y, z = self.fill_gaps(*self.decimate()) | |
|
|
429 | ||
|
|
430 | for n, ax in enumerate(self.axes): | |
|
|
431 | self.zmax = self.zmax if self.zmax is not None else numpy.max( | |
|
|
432 | self.z[n]) | |
|
|
433 | self.zmin = self.zmin if self.zmin is not None else numpy.min( | |
|
|
434 | self.z[n]) | |
|
|
435 | ||
|
|
436 | if ax.firsttime: | |
|
|
437 | if self.zlimits is not None: | |
|
|
438 | self.zmin, self.zmax = self.zlimits[n] | |
|
|
439 | ||
|
|
440 | ax.plt = ax.pcolormesh(x, y, z[n].T * self.factors[n], | |
|
|
441 | vmin=self.zmin, | |
|
|
442 | vmax=self.zmax, | |
|
|
443 | cmap=self.cmaps[n] | |
|
|
444 | ) | |
|
|
445 | else: | |
|
|
446 | if self.zlimits is not None: | |
|
|
447 | self.zmin, self.zmax = self.zlimits[n] | |
|
|
448 | ax.plt.remove() | |
|
|
449 | ax.plt = ax.pcolormesh(x, y, z[n].T * self.factors[n], | |
|
|
450 | vmin=self.zmin, | |
|
|
451 | vmax=self.zmax, | |
|
|
452 | cmap=self.cmaps[n] | |
|
|
453 | ) | |
|
|
454 | ||
|
|
455 | class AverageDriftsPlot_v2(Plot): | |
|
|
456 | ''' | |
|
|
457 | Plot for average 150 Km echoes | |
|
|
458 | ''' | |
|
|
459 | ||
|
|
460 | CODE = 'average' | |
|
|
461 | plot_type = 'scatterbuffer' | |
|
|
462 | ||
|
|
463 | def setup(self): | |
|
|
464 | self.xaxis = 'time' | |
|
|
465 | self.ncols = 1 | |
|
|
466 | ||
|
|
467 | self.nplots = 2 | |
|
|
468 | self.nrows = 2 | |
|
|
469 | ||
|
|
470 | self.ylabel = 'Velocity\nm/s' | |
|
|
471 | self.xlabel = 'Local time' | |
|
|
472 | #self.titles = ['VERTICAL VELOCITY: AVERAGE AND ERRORS', 'ZONAL VELOCITY: AVERAGE AND ERRORS'] | |
|
|
473 | self.titles = ['VERTICAL VELOCITY: AVERAGE', 'ZONAL VELOCITY: AVERAGE'] | |
|
|
474 | ||
|
|
475 | self.colorbar = False | |
|
|
476 | self.plots_adjust.update({'hspace':0.5, 'left': 0.1, 'bottom': 0.1, 'right':0.95, 'top': 0.95 }) | |
|
|
477 | ||
|
|
478 | ||
|
|
479 | def update(self, dataOut): | |
|
|
480 | ||
|
|
481 | data = {} | |
|
|
482 | meta = {} | |
|
|
483 | ||
|
|
484 | #data['average']= numpy.nanmean(dataOut.data_output[1:3,:], axis=1) # VERTICAL, ZONAL | |
|
|
485 | data['average']= numpy.nanmean(dataOut.data_output[1:3,:], axis=1) # VERTICAL, ZONAL | |
|
|
486 | data['error']= numpy.nanmean(dataOut.data_output[3:,:], axis=1) # ERROR VERTICAL, ERROR ZONAL | |
|
|
487 | meta['yrange'] = numpy.array([]) | |
|
|
488 | ||
|
|
489 | return data, meta | |
|
|
490 | ||
|
|
491 | def plot(self): | |
|
|
492 | ||
|
|
493 | self.x = self.data.times | |
|
|
494 | #self.xmin = self.data.min_time | |
|
|
495 | #self.xmax = self.xmin + self.xrange * 60 * 60 | |
|
|
496 | self.y = self.data['average'] | |
|
|
497 | print('self.y:', self.y.shape) | |
|
|
498 | self.y_error = self.data['error'] | |
|
|
499 | print('self.y_error:', self.y_error.shape) | |
|
|
500 | ||
|
|
501 | for n, ax in enumerate(self.axes): | |
|
|
502 | if ax.firsttime: | |
|
|
503 | self.ymin = self.ymin if self.ymin is not None else -50 | |
|
|
504 | self.ymax = self.ymax if self.ymax is not None else 50 | |
|
|
505 | self.axes[n].plot(self.x, self.y[n], c='r', ls=':', lw=1) | |
|
|
506 | else: | |
|
|
507 | self.axes[n].lines[0].set_data(self.x, self.y[n]) | |
|
|
508 | ''' | |
|
|
509 | for n, ax in enumerate(self.axes): | |
|
|
510 | ||
|
|
511 | if ax.firsttime: | |
|
|
512 | self.ymin = self.ymin if self.ymin is not None else -50 | |
|
|
513 | self.ymax = self.ymax if self.ymax is not None else 50 | |
|
|
514 | ax.scatter(self.x, self.y[n], c='g', s=0.8) | |
|
|
515 | #ax.errorbar(self.x, self.y[n], yerr = self.y_error[n,:], ecolor='r', elinewidth=0.2, fmt='|') | |
|
|
516 | else: | |
|
|
517 | ax.scatter(self.x, self.y[n], c='g', s=0.8) | |
|
|
518 | #ax.errorbar(self.x, self.y[n], yerr = self.y_error[n,:], ecolor='r', elinewidth=0.2, fmt='|') | |
|
|
519 | ''' | |
|
|
520 | class AverageDriftsPlot_bck(Plot): | |
|
|
521 | ''' | |
|
|
522 | Plot for average 150 Km echoes | |
|
|
523 | ''' | |
|
|
524 | CODE = 'average' | |
|
|
525 | plot_type = 'scatterbuffer' | |
|
|
526 | ||
|
|
527 | def setup(self): | |
|
|
528 | self.xaxis = 'time' | |
|
|
529 | self.ncols = 1 | |
|
|
530 | self.nplots = 2 | |
|
|
531 | self.nrows = 2 | |
|
|
532 | self.ylabel = 'Velocity\nm/s' | |
|
|
533 | self.xlabel = 'Time' | |
|
|
534 | self.titles = ['VERTICAL VELOCITY: AVERAGE', 'ZONAL VELOCITY: AVERAGE'] | |
|
|
535 | self.colorbar = False | |
|
|
536 | self.plots_adjust.update({'hspace':0.5, 'left': 0.1, 'bottom': 0.1, 'right':0.95, 'top': 0.95 }) | |
|
|
537 | ||
|
|
538 | ||
|
|
539 | ||
|
|
540 | def update(self, dataOut): | |
|
|
541 | ||
|
|
542 | data = {} | |
|
|
543 | meta = {} | |
|
|
544 | data['average']= numpy.nanmean(dataOut.data_output[1:3,:], axis=1) # VERTICAL, ZONAL | |
|
|
545 | meta['yrange'] = numpy.array([]) | |
|
|
546 | ||
|
|
547 | return data, meta | |
|
|
548 | ||
|
|
549 | def plot(self): | |
|
|
550 | ||
|
|
551 | self.x = self.data.times | |
|
|
552 | self.y = self.data['average'] | |
|
|
553 | ||
|
|
554 | for n, ax in enumerate(self.axes): | |
|
|
555 | if ax.firsttime: | |
|
|
556 | ||
|
|
557 | if self.zlimits is not None: | |
|
|
558 | self.axes[n].set_ylim(self.zlimits[n]) | |
|
|
559 | self.axes[n].plot(self.x, self.y[n], c='r', ls='-', lw=1) | |
|
|
560 | else: | |
|
|
561 | ||
|
|
562 | if self.zlimits is not None: | |
|
|
563 | ax.set_ylim((self.zlimits[n])) | |
|
|
564 | self.axes[n].lines[0].set_data(self.x, self.y[n]) | |
|
|
565 | ||
|
|
566 | class AverageDriftsPlot(Plot): | |
|
|
567 | ''' | |
|
|
568 | Plot for average 150 Km echoes | |
|
|
569 | ''' | |
|
|
570 | CODE = 'average' | |
|
|
571 | plot_type = 'scatterbuffer' | |
|
|
572 | ||
|
|
573 | def setup(self): | |
|
|
574 | self.xaxis = 'time' | |
|
|
575 | self.ncols = 1 | |
|
|
576 | self.nplots = 2 | |
|
|
577 | self.nrows = 2 | |
|
|
578 | self.ylabel = 'Velocity\nm/s' | |
|
|
579 | self.xlabel = 'Time' | |
|
|
580 | self.titles = ['VERTICAL VELOCITY: AVERAGE', 'ZONAL VELOCITY: AVERAGE'] | |
|
|
581 | self.colorbar = False | |
|
|
582 | self.plots_adjust.update({'hspace':0.5, 'left': 0.1, 'bottom': 0.1, 'right':0.95, 'top': 0.95 }) | |
|
|
583 | ||
|
|
584 | ||
|
|
585 | ||
|
|
586 | def update(self, dataOut): | |
|
|
587 | ||
|
|
588 | data = {} | |
|
|
589 | meta = {} | |
|
|
590 | data['average']= dataOut.avg_output[0:2] # VERTICAL, ZONAL velocities | |
|
|
591 | meta['yrange'] = numpy.array([]) | |
|
|
592 | ||
|
|
593 | return data, meta | |
|
|
594 | ||
|
|
595 | def plot(self): | |
|
|
596 | ||
|
|
597 | self.x = self.data.times | |
|
|
598 | self.y = self.data['average'] | |
|
|
599 | ||
|
|
600 | for n, ax in enumerate(self.axes): | |
|
|
601 | ||
|
|
602 | if ax.firsttime: | |
|
|
603 | ||
|
|
604 | if self.zlimits is not None: | |
|
|
605 | ax.set_ylim((self.zlimits[n])) | |
|
|
606 | self.axes[n].plot(self.x, self.y[n], c='r', ls='-', lw=1) | |
|
|
607 | else: | |
|
|
608 | ||
|
|
609 | if self.zlimits is not None: | |
|
|
610 | ax.set_ylim((self.zlimits[n])) | |
|
|
611 | self.axes[n].lines[0].set_data(self.x, self.y[n]) | |
|
|
612 | ||
General Comments 0
You need to be logged in to leave comments.
Login now