jroplot_correlation.py
186 lines
| 5.9 KiB
| text/x-python
|
PythonLexer
Julio Valdez
|
r502 | import os | |
| import datetime | |||
| import numpy | |||
| import copy | |||
José Chávez
|
r954 | from schainpy.model import * | |
George Yong
|
r1167 | from .figure import Figure, isRealtime | |
|
Julio Valdez
|
r502 | ||
| class CorrelationPlot(Figure): | |||
| isConfig = None | |||
| __nsubplots = None | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | WIDTHPROF = None | |
| HEIGHTPROF = None | |||
| PREFIX = 'corr' | |||
|
José Chávez
|
r897 | ||
| def __init__(self, **kwargs): | |||
| Figure.__init__(self, **kwargs) | |||
|
Julio Valdez
|
r502 | self.isConfig = False | |
| self.__nsubplots = 1 | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | self.WIDTH = 280 | |
| self.HEIGHT = 250 | |||
| self.WIDTHPROF = 120 | |||
| self.HEIGHTPROF = 0 | |||
| self.counter_imagwr = 0 | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | self.PLOT_CODE = 1 | |
| self.FTP_WEI = None | |||
| self.EXP_CODE = None | |||
| self.SUB_EXP_CODE = None | |||
| self.PLOT_POS = None | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | def getSubplots(self): | |
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | ncol = int(numpy.sqrt(self.nplots)+0.9) | |
| nrow = int(self.nplots*1./ncol + 0.9) | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | return nrow, ncol | |
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | def setup(self, id, nplots, wintitle, showprofile=False, show=True): | |
|
José Chávez
|
r897 | ||
| showprofile = False | |||
|
Julio Valdez
|
r502 | self.__showprofile = showprofile | |
| self.nplots = nplots | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | ncolspan = 1 | |
| colspan = 1 | |||
| if showprofile: | |||
| ncolspan = 3 | |||
| colspan = 2 | |||
| self.__nsubplots = 2 | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | self.createFigure(id = id, | |
| wintitle = wintitle, | |||
| widthplot = self.WIDTH + self.WIDTHPROF, | |||
| heightplot = self.HEIGHT + self.HEIGHTPROF, | |||
| show=show) | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | nrow, ncol = self.getSubplots() | |
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | counter = 0 | |
| for y in range(nrow): | |||
| for x in range(ncol): | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | if counter >= self.nplots: | |
| break | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan, colspan, 1) | |
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | if showprofile: | |
| self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan+colspan, 1, 1) | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | counter += 1 | |
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | def run(self, dataOut, id, wintitle="", channelList=None, showprofile=False, | |
| xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None, | |||
Miguel Valdez
|
r568 | save=False, figpath='./', figfile=None, show=True, ftp=False, wr_period=1, | |
|
Julio Valdez
|
r502 | server=None, folder=None, username=None, password=None, | |
| ftp_wei=0, exp_code=0, sub_exp_code=0, plot_pos=0, realtime=False): | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | """ | |
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | Input: | |
| dataOut : | |||
| id : | |||
| wintitle : | |||
| channelList : | |||
| showProfile : | |||
| xmin : None, | |||
| xmax : None, | |||
| ymin : None, | |||
| ymax : None, | |||
| zmin : None, | |||
| zmax : None | |||
| """ | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | if dataOut.flagNoData: | |
| return None | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | if realtime: | |
| if not(isRealtime(utcdatatime = dataOut.utctime)): | |||
|
George Yong
|
r1167 | print('Skipping this plot function') | |
|
Julio Valdez
|
r502 | return | |
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | if channelList == None: | |
| channelIndexList = dataOut.channelIndexList | |||
| else: | |||
| channelIndexList = [] | |||
| for channel in channelList: | |||
| if channel not in dataOut.channelList: | |||
|
George Yong
|
r1167 | raise ValueError("Channel %d is not in dataOut.channelList") | |
|
Julio Valdez
|
r502 | channelIndexList.append(dataOut.channelList.index(channel)) | |
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | factor = dataOut.normFactor | |
| lenfactor = factor.shape[1] | |||
| x = dataOut.getLagTRange(1) | |||
| y = dataOut.getHeiRange() | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | z = copy.copy(dataOut.data_corr[:,:,0,:]) | |
| for i in range(dataOut.data_corr.shape[0]): | |||
|
José Chávez
|
r897 | z[i,:,:] = z[i,:,:]/factor[i,:] | |
|
Julio Valdez
|
r502 | zdB = numpy.abs(z) | |
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | avg = numpy.average(z, axis=1) | |
| # avg = numpy.nanmean(z, axis=1) | |||
| # noise = dataOut.noise/factor | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | #thisDatetime = dataOut.datatime | |
|
Miguel Valdez
|
r568 | thisDatetime = datetime.datetime.utcfromtimestamp(dataOut.getTimeRange()[0]) | |
|
José Chávez
|
r897 | title = wintitle + " Correlation" | |
|
Julio Valdez
|
r502 | xlabel = "Lag T (s)" | |
| ylabel = "Range (Km)" | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | if not self.isConfig: | |
|
José Chávez
|
r897 | ||
| nplots = dataOut.data_corr.shape[0] | |||
|
Julio Valdez
|
r502 | self.setup(id=id, | |
| nplots=nplots, | |||
| wintitle=wintitle, | |||
| showprofile=showprofile, | |||
| show=show) | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | if xmin == None: xmin = numpy.nanmin(x) | |
| if xmax == None: xmax = numpy.nanmax(x) | |||
| if ymin == None: ymin = numpy.nanmin(y) | |||
| if ymax == None: ymax = numpy.nanmax(y) | |||
| if zmin == None: zmin = 0 | |||
| if zmax == None: zmax = 1 | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | self.FTP_WEI = ftp_wei | |
| self.EXP_CODE = exp_code | |||
| self.SUB_EXP_CODE = sub_exp_code | |||
| self.PLOT_POS = plot_pos | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | self.isConfig = True | |
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | self.setWinTitle(title) | |
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | for i in range(self.nplots): | |
| str_datetime = '%s %s'%(thisDatetime.strftime("%Y/%m/%d"),thisDatetime.strftime("%H:%M:%S")) | |||
|
Miguel Valdez
|
r713 | title = "Channel %d and %d: : %s" %(dataOut.pairsList[i][0],dataOut.pairsList[i][1] , str_datetime) | |
|
Julio Valdez
|
r502 | axes = self.axesList[i*self.__nsubplots] | |
| axes.pcolor(x, y, zdB[i,:,:], | |||
| xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax, zmin=zmin, zmax=zmax, | |||
| xlabel=xlabel, ylabel=ylabel, title=title, | |||
| ticksize=9, cblabel='') | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | # if self.__showprofile: | |
| # axes = self.axesList[i*self.__nsubplots +1] | |||
| # axes.pline(avgdB[i], y, | |||
| # xmin=zmin, xmax=zmax, ymin=ymin, ymax=ymax, | |||
| # xlabel='dB', ylabel='', title='', | |||
| # ytick_visible=False, | |||
| # grid='x') | |||
|
José Chávez
|
r897 | # | |
|
Julio Valdez
|
r502 | # noiseline = numpy.repeat(noisedB[i], len(y)) | |
| # axes.addpline(noiseline, y, idline=1, color="black", linestyle="dashed", lw=2) | |||
|
José Chávez
|
r897 | ||
|
Julio Valdez
|
r502 | self.draw() | |
|
José Chávez
|
r897 | ||
|
Miguel Valdez
|
r573 | self.save(figpath=figpath, | |
| figfile=figfile, | |||
| save=save, | |||
| ftp=ftp, | |||
| wr_period=wr_period, | |||
|
George Yong
|
r1167 | thisDatetime=thisDatetime) |