|
|
'''
|
|
|
Created on Jul 9, 2014
|
|
|
|
|
|
@author: roj-idl71
|
|
|
'''
|
|
|
import os
|
|
|
import datetime
|
|
|
import numpy
|
|
|
|
|
|
from .figure import Figure, isRealtime, isTimeInHourRange
|
|
|
from .plotting_codes import *
|
|
|
from schainpy.model.proc.jroproc_base import MPDecorator
|
|
|
|
|
|
from schainpy.utils import log
|
|
|
|
|
|
@MPDecorator
|
|
|
class SpectraPlot_(Figure):
|
|
|
|
|
|
isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
WIDTHPROF = None
|
|
|
HEIGHTPROF = None
|
|
|
PREFIX = 'spc'
|
|
|
|
|
|
def __init__(self):
|
|
|
Figure.__init__(self)
|
|
|
self.isConfig = False
|
|
|
self.__nsubplots = 1
|
|
|
self.WIDTH = 250
|
|
|
self.HEIGHT = 250
|
|
|
self.WIDTHPROF = 120
|
|
|
self.HEIGHTPROF = 0
|
|
|
self.counter_imagwr = 0
|
|
|
|
|
|
self.PLOT_CODE = SPEC_CODE
|
|
|
|
|
|
self.FTP_WEI = None
|
|
|
self.EXP_CODE = None
|
|
|
self.SUB_EXP_CODE = None
|
|
|
self.PLOT_POS = None
|
|
|
|
|
|
self.__xfilter_ena = False
|
|
|
self.__yfilter_ena = False
|
|
|
|
|
|
self.indice=1
|
|
|
|
|
|
def getSubplots(self):
|
|
|
|
|
|
ncol = int(numpy.sqrt(self.nplots)+0.9)
|
|
|
nrow = int(self.nplots*1./ncol + 0.9)
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, id, nplots, wintitle, showprofile=True, show=True):
|
|
|
|
|
|
self.__showprofile = showprofile
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 1
|
|
|
colspan = 1
|
|
|
if showprofile:
|
|
|
ncolspan = 3
|
|
|
colspan = 2
|
|
|
self.__nsubplots = 2
|
|
|
|
|
|
self.createFigure(id = id,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = self.WIDTH + self.WIDTHPROF,
|
|
|
heightplot = self.HEIGHT + self.HEIGHTPROF,
|
|
|
show=show)
|
|
|
|
|
|
nrow, ncol = self.getSubplots()
|
|
|
|
|
|
counter = 0
|
|
|
for y in range(nrow):
|
|
|
for x in range(ncol):
|
|
|
|
|
|
if counter >= self.nplots:
|
|
|
break
|
|
|
|
|
|
self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan, colspan, 1)
|
|
|
|
|
|
if showprofile:
|
|
|
self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan+colspan, 1, 1)
|
|
|
|
|
|
counter += 1
|
|
|
|
|
|
def run(self, dataOut, id, wintitle="", channelList=None, showprofile=True,
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None,
|
|
|
save=False, figpath='./', figfile=None, show=True, ftp=False, wr_period=1,
|
|
|
server=None, folder=None, username=None, password=None,
|
|
|
ftp_wei=0, exp_code=0, sub_exp_code=0, plot_pos=0, realtime=False,
|
|
|
xaxis="frequency", colormap='jet', normFactor=None):
|
|
|
|
|
|
"""
|
|
|
|
|
|
Input:
|
|
|
dataOut :
|
|
|
id :
|
|
|
wintitle :
|
|
|
channelList :
|
|
|
showProfile :
|
|
|
xmin : None,
|
|
|
xmax : None,
|
|
|
ymin : None,
|
|
|
ymax : None,
|
|
|
zmin : None,
|
|
|
zmax : None
|
|
|
"""
|
|
|
if dataOut.flagNoData:
|
|
|
return dataOut
|
|
|
|
|
|
if realtime:
|
|
|
if not(isRealtime(utcdatatime = dataOut.utctime)):
|
|
|
print('Skipping this plot function')
|
|
|
return
|
|
|
|
|
|
if channelList == None:
|
|
|
channelIndexList = dataOut.channelIndexList
|
|
|
else:
|
|
|
channelIndexList = []
|
|
|
for channel in channelList:
|
|
|
if channel not in dataOut.channelList:
|
|
|
raise ValueError("Channel %d is not in dataOut.channelList" %channel)
|
|
|
channelIndexList.append(dataOut.channelList.index(channel))
|
|
|
|
|
|
if normFactor is None:
|
|
|
factor = dataOut.normFactor
|
|
|
else:
|
|
|
factor = normFactor
|
|
|
if xaxis == "frequency":
|
|
|
x = dataOut.getFreqRange(1)/1000.
|
|
|
xlabel = "Frequency (kHz)"
|
|
|
|
|
|
elif xaxis == "time":
|
|
|
x = dataOut.getAcfRange(1)
|
|
|
xlabel = "Time (ms)"
|
|
|
|
|
|
else:
|
|
|
x = dataOut.getVelRange(1)
|
|
|
xlabel = "Velocity (m/s)"
|
|
|
|
|
|
ylabel = "Range (km)"
|
|
|
|
|
|
y = dataOut.getHeiRange()
|
|
|
z = dataOut.data_spc/factor
|
|
|
z = numpy.where(numpy.isfinite(z), z, numpy.NAN)
|
|
|
zdB = 10*numpy.log10(z)
|
|
|
|
|
|
avg = numpy.average(z, axis=1)
|
|
|
avgdB = 10*numpy.log10(avg)
|
|
|
|
|
|
noise = dataOut.getNoise()/factor
|
|
|
noisedB = 10*numpy.log10(noise)
|
|
|
|
|
|
thisDatetime = datetime.datetime.utcfromtimestamp(dataOut.getTimeRange()[0])
|
|
|
title = wintitle + " Spectra"
|
|
|
|
|
|
if ((dataOut.azimuth!=None) and (dataOut.zenith!=None)):
|
|
|
title = title + '_' + 'azimuth,zenith=%2.2f,%2.2f'%(dataOut.azimuth, dataOut.zenith)
|
|
|
|
|
|
if not self.isConfig:
|
|
|
|
|
|
nplots = len(channelIndexList)
|
|
|
|
|
|
self.setup(id=id,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
showprofile=showprofile,
|
|
|
show=show)
|
|
|
|
|
|
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 = numpy.floor(numpy.nanmin(noisedB)) - 3
|
|
|
if zmax == None: zmax = numpy.ceil(numpy.nanmax(avgdB)) + 3
|
|
|
|
|
|
self.FTP_WEI = ftp_wei
|
|
|
self.EXP_CODE = exp_code
|
|
|
self.SUB_EXP_CODE = sub_exp_code
|
|
|
self.PLOT_POS = plot_pos
|
|
|
|
|
|
self.isConfig = True
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
for i in range(self.nplots):
|
|
|
index = channelIndexList[i]
|
|
|
str_datetime = '%s %s'%(thisDatetime.strftime("%Y/%m/%d"),thisDatetime.strftime("%H:%M:%S"))
|
|
|
title = "Channel %d: %4.2fdB: %s" %(dataOut.channelList[index], noisedB[index], str_datetime)
|
|
|
if len(dataOut.beam.codeList) != 0:
|
|
|
title = "Ch%d:%4.2fdB,%2.2f,%2.2f:%s" %(dataOut.channelList[index], noisedB[index], dataOut.beam.azimuthList[index], dataOut.beam.zenithList[index], str_datetime)
|
|
|
|
|
|
axes = self.axesList[i*self.__nsubplots]
|
|
|
axes.pcolor(x, y, zdB[index,:,:],
|
|
|
xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax, zmin=zmin, zmax=zmax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title, colormap=colormap,
|
|
|
ticksize=9, cblabel='')
|
|
|
|
|
|
if self.__showprofile:
|
|
|
axes = self.axesList[i*self.__nsubplots +1]
|
|
|
axes.pline(avgdB[index,:], y,
|
|
|
xmin=zmin, xmax=zmax, ymin=ymin, ymax=ymax,
|
|
|
xlabel='dB', ylabel='', title='',
|
|
|
ytick_visible=False,
|
|
|
grid='x')
|
|
|
|
|
|
noiseline = numpy.repeat(noisedB[index], len(y))
|
|
|
axes.addpline(noiseline, y, idline=1, color="black", linestyle="dashed", lw=2)
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
if figfile == None:
|
|
|
str_datetime = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
name = str_datetime
|
|
|
if ((dataOut.azimuth!=None) and (dataOut.zenith!=None)):
|
|
|
name = name + '_az' + '_%2.2f'%(dataOut.azimuth) + '_zn' + '_%2.2f'%(dataOut.zenith)
|
|
|
figfile = self.getFilename(name)
|
|
|
|
|
|
self.save(figpath=figpath,
|
|
|
figfile=figfile,
|
|
|
save=save,
|
|
|
ftp=ftp,
|
|
|
wr_period=wr_period,
|
|
|
thisDatetime=thisDatetime)
|
|
|
|
|
|
|
|
|
return dataOut
|
|
|
|
|
|
@MPDecorator
|
|
|
class CrossSpectraPlot_(Figure):
|
|
|
|
|
|
isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
WIDTH = None
|
|
|
HEIGHT = None
|
|
|
WIDTHPROF = None
|
|
|
HEIGHTPROF = None
|
|
|
PREFIX = 'cspc'
|
|
|
|
|
|
def __init__(self):
|
|
|
Figure.__init__(self)
|
|
|
self.isConfig = False
|
|
|
self.__nsubplots = 4
|
|
|
self.counter_imagwr = 0
|
|
|
self.WIDTH = 250
|
|
|
self.HEIGHT = 250
|
|
|
self.WIDTHPROF = 0
|
|
|
self.HEIGHTPROF = 0
|
|
|
|
|
|
self.PLOT_CODE = CROSS_CODE
|
|
|
self.FTP_WEI = None
|
|
|
self.EXP_CODE = None
|
|
|
self.SUB_EXP_CODE = None
|
|
|
self.PLOT_POS = None
|
|
|
|
|
|
self.indice=0
|
|
|
|
|
|
def getSubplots(self):
|
|
|
|
|
|
ncol = 4
|
|
|
nrow = self.nplots
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, id, nplots, wintitle, showprofile=True, show=True):
|
|
|
|
|
|
self.__showprofile = showprofile
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 1
|
|
|
colspan = 1
|
|
|
|
|
|
self.createFigure(id = id,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = self.WIDTH + self.WIDTHPROF,
|
|
|
heightplot = self.HEIGHT + self.HEIGHTPROF,
|
|
|
show=True)
|
|
|
|
|
|
nrow, ncol = self.getSubplots()
|
|
|
|
|
|
counter = 0
|
|
|
for y in range(nrow):
|
|
|
for x in range(ncol):
|
|
|
self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan, colspan, 1)
|
|
|
|
|
|
counter += 1
|
|
|
|
|
|
def run(self, dataOut, id, wintitle="", pairsList=None,
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None,
|
|
|
coh_min=None, coh_max=None, phase_min=None, phase_max=None,
|
|
|
save=False, figpath='./', figfile=None, ftp=False, wr_period=1,
|
|
|
power_cmap='jet', coherence_cmap='jet', phase_cmap='RdBu_r', show=True,
|
|
|
server=None, folder=None, username=None, password=None,
|
|
|
ftp_wei=0, exp_code=0, sub_exp_code=0, plot_pos=0, normFactor=None,
|
|
|
xaxis='frequency'):
|
|
|
|
|
|
"""
|
|
|
|
|
|
Input:
|
|
|
dataOut :
|
|
|
id :
|
|
|
wintitle :
|
|
|
channelList :
|
|
|
showProfile :
|
|
|
xmin : None,
|
|
|
xmax : None,
|
|
|
ymin : None,
|
|
|
ymax : None,
|
|
|
zmin : None,
|
|
|
zmax : None
|
|
|
"""
|
|
|
|
|
|
if dataOut.flagNoData:
|
|
|
return dataOut
|
|
|
|
|
|
if pairsList == None:
|
|
|
pairsIndexList = dataOut.pairsIndexList
|
|
|
else:
|
|
|
pairsIndexList = []
|
|
|
for pair in pairsList:
|
|
|
if pair not in dataOut.pairsList:
|
|
|
raise ValueError("Pair %s is not in dataOut.pairsList" %str(pair))
|
|
|
pairsIndexList.append(dataOut.pairsList.index(pair))
|
|
|
|
|
|
if not pairsIndexList:
|
|
|
return
|
|
|
|
|
|
if len(pairsIndexList) > 4:
|
|
|
pairsIndexList = pairsIndexList[0:4]
|
|
|
|
|
|
if normFactor is None:
|
|
|
factor = dataOut.normFactor
|
|
|
else:
|
|
|
factor = normFactor
|
|
|
x = dataOut.getVelRange(1)
|
|
|
y = dataOut.getHeiRange()
|
|
|
z = dataOut.data_spc[:,:,:]/factor
|
|
|
z = numpy.where(numpy.isfinite(z), z, numpy.NAN)
|
|
|
|
|
|
noise = dataOut.noise/factor
|
|
|
|
|
|
zdB = 10*numpy.log10(z)
|
|
|
noisedB = 10*numpy.log10(noise)
|
|
|
|
|
|
if coh_min == None:
|
|
|
coh_min = 0.0
|
|
|
if coh_max == None:
|
|
|
coh_max = 1.0
|
|
|
|
|
|
if phase_min == None:
|
|
|
phase_min = -180
|
|
|
if phase_max == None:
|
|
|
phase_max = 180
|
|
|
|
|
|
#thisDatetime = dataOut.datatime
|
|
|
thisDatetime = datetime.datetime.utcfromtimestamp(dataOut.getTimeRange()[0])
|
|
|
title = wintitle + " Cross-Spectra: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
# xlabel = "Velocity (m/s)"
|
|
|
ylabel = "Range (Km)"
|
|
|
|
|
|
if xaxis == "frequency":
|
|
|
x = dataOut.getFreqRange(1)/1000.
|
|
|
xlabel = "Frequency (kHz)"
|
|
|
|
|
|
elif xaxis == "time":
|
|
|
x = dataOut.getAcfRange(1)
|
|
|
xlabel = "Time (ms)"
|
|
|
|
|
|
else:
|
|
|
x = dataOut.getVelRange(1)
|
|
|
xlabel = "Velocity (m/s)"
|
|
|
|
|
|
if not self.isConfig:
|
|
|
|
|
|
nplots = len(pairsIndexList)
|
|
|
|
|
|
self.setup(id=id,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
showprofile=False,
|
|
|
show=show)
|
|
|
|
|
|
avg = numpy.abs(numpy.average(z, axis=1))
|
|
|
avgdB = 10*numpy.log10(avg)
|
|
|
|
|
|
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 = numpy.floor(numpy.nanmin(noisedB)) - 3
|
|
|
if zmax == None: zmax = numpy.ceil(numpy.nanmax(avgdB)) + 3
|
|
|
|
|
|
self.FTP_WEI = ftp_wei
|
|
|
self.EXP_CODE = exp_code
|
|
|
self.SUB_EXP_CODE = sub_exp_code
|
|
|
self.PLOT_POS = plot_pos
|
|
|
|
|
|
self.isConfig = True
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
|
|
|
for i in range(self.nplots):
|
|
|
pair = dataOut.pairsList[pairsIndexList[i]]
|
|
|
|
|
|
chan_index0 = dataOut.channelList.index(pair[0])
|
|
|
chan_index1 = dataOut.channelList.index(pair[1])
|
|
|
|
|
|
str_datetime = '%s %s'%(thisDatetime.strftime("%Y/%m/%d"),thisDatetime.strftime("%H:%M:%S"))
|
|
|
title = "Ch%d: %4.2fdB: %s" %(pair[0], noisedB[chan_index0], str_datetime)
|
|
|
zdB = 10.*numpy.log10(dataOut.data_spc[chan_index0,:,:]/factor)
|
|
|
axes0 = self.axesList[i*self.__nsubplots]
|
|
|
axes0.pcolor(x, y, zdB,
|
|
|
xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax, zmin=zmin, zmax=zmax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title,
|
|
|
ticksize=9, colormap=power_cmap, cblabel='')
|
|
|
|
|
|
title = "Ch%d: %4.2fdB: %s" %(pair[1], noisedB[chan_index1], str_datetime)
|
|
|
zdB = 10.*numpy.log10(dataOut.data_spc[chan_index1,:,:]/factor)
|
|
|
axes0 = self.axesList[i*self.__nsubplots+1]
|
|
|
axes0.pcolor(x, y, zdB,
|
|
|
xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax, zmin=zmin, zmax=zmax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title,
|
|
|
ticksize=9, colormap=power_cmap, cblabel='')
|
|
|
|
|
|
coherenceComplex = dataOut.data_cspc[pairsIndexList[i],:,:] / numpy.sqrt( dataOut.data_spc[chan_index0,:,:]*dataOut.data_spc[chan_index1,:,:] )
|
|
|
coherence = numpy.abs(coherenceComplex)
|
|
|
# phase = numpy.arctan(-1*coherenceComplex.imag/coherenceComplex.real)*180/numpy.pi
|
|
|
phase = numpy.arctan2(coherenceComplex.imag, coherenceComplex.real)*180/numpy.pi
|
|
|
|
|
|
title = "Coherence Ch%d * Ch%d" %(pair[0], pair[1])
|
|
|
axes0 = self.axesList[i*self.__nsubplots+2]
|
|
|
axes0.pcolor(x, y, coherence,
|
|
|
xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax, zmin=coh_min, zmax=coh_max,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title,
|
|
|
ticksize=9, colormap=coherence_cmap, cblabel='')
|
|
|
|
|
|
title = "Phase Ch%d * Ch%d" %(pair[0], pair[1])
|
|
|
axes0 = self.axesList[i*self.__nsubplots+3]
|
|
|
axes0.pcolor(x, y, phase,
|
|
|
xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax, zmin=phase_min, zmax=phase_max,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title,
|
|
|
ticksize=9, colormap=phase_cmap, cblabel='')
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
self.save(figpath=figpath,
|
|
|
figfile=figfile,
|
|
|
save=save,
|
|
|
ftp=ftp,
|
|
|
wr_period=wr_period,
|
|
|
thisDatetime=thisDatetime)
|
|
|
|
|
|
return dataOut
|
|
|
|
|
|
@MPDecorator
|
|
|
class RTIPlot_(Figure):
|
|
|
|
|
|
__isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
WIDTHPROF = None
|
|
|
HEIGHTPROF = None
|
|
|
PREFIX = 'rti'
|
|
|
|
|
|
def __init__(self):
|
|
|
|
|
|
Figure.__init__(self)
|
|
|
self.timerange = None
|
|
|
self.isConfig = False
|
|
|
self.__nsubplots = 1
|
|
|
|
|
|
self.WIDTH = 800
|
|
|
self.HEIGHT = 250
|
|
|
self.WIDTHPROF = 120
|
|
|
self.HEIGHTPROF = 0
|
|
|
self.counter_imagwr = 0
|
|
|
|
|
|
self.PLOT_CODE = RTI_CODE
|
|
|
|
|
|
self.FTP_WEI = None
|
|
|
self.EXP_CODE = None
|
|
|
self.SUB_EXP_CODE = None
|
|
|
self.PLOT_POS = None
|
|
|
self.tmin = None
|
|
|
self.tmax = None
|
|
|
|
|
|
self.xmin = None
|
|
|
self.xmax = None
|
|
|
|
|
|
self.figfile = None
|
|
|
|
|
|
def getSubplots(self):
|
|
|
|
|
|
ncol = 1
|
|
|
nrow = self.nplots
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, id, nplots, wintitle, showprofile=True, show=True):
|
|
|
|
|
|
self.__showprofile = showprofile
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 1
|
|
|
colspan = 1
|
|
|
if showprofile:
|
|
|
ncolspan = 7
|
|
|
colspan = 6
|
|
|
self.__nsubplots = 2
|
|
|
|
|
|
self.createFigure(id = id,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = self.WIDTH + self.WIDTHPROF,
|
|
|
heightplot = self.HEIGHT + self.HEIGHTPROF,
|
|
|
show=show)
|
|
|
|
|
|
nrow, ncol = self.getSubplots()
|
|
|
|
|
|
counter = 0
|
|
|
for y in range(nrow):
|
|
|
for x in range(ncol):
|
|
|
|
|
|
if counter >= self.nplots:
|
|
|
break
|
|
|
|
|
|
self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan, colspan, 1)
|
|
|
|
|
|
if showprofile:
|
|
|
self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan+colspan, 1, 1)
|
|
|
|
|
|
counter += 1
|
|
|
|
|
|
def run(self, dataOut, id, wintitle="", channelList=None, showprofile='True',
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None,
|
|
|
timerange=None, colormap='jet',
|
|
|
save=False, figpath='./', lastone=0,figfile=None, ftp=False, wr_period=1, show=True,
|
|
|
server=None, folder=None, username=None, password=None,
|
|
|
ftp_wei=0, exp_code=0, sub_exp_code=0, plot_pos=0, normFactor=None, HEIGHT=None):
|
|
|
|
|
|
"""
|
|
|
|
|
|
Input:
|
|
|
dataOut :
|
|
|
id :
|
|
|
wintitle :
|
|
|
channelList :
|
|
|
showProfile :
|
|
|
xmin : None,
|
|
|
xmax : None,
|
|
|
ymin : None,
|
|
|
ymax : None,
|
|
|
zmin : None,
|
|
|
zmax : None
|
|
|
"""
|
|
|
if dataOut.flagNoData:
|
|
|
return dataOut
|
|
|
|
|
|
#colormap = kwargs.get('colormap', 'jet')
|
|
|
if HEIGHT is not None:
|
|
|
self.HEIGHT = HEIGHT
|
|
|
|
|
|
if not isTimeInHourRange(dataOut.datatime, xmin, xmax):
|
|
|
return
|
|
|
|
|
|
if channelList == None:
|
|
|
channelIndexList = dataOut.channelIndexList
|
|
|
else:
|
|
|
channelIndexList = []
|
|
|
for channel in channelList:
|
|
|
if channel not in dataOut.channelList:
|
|
|
raise ValueError("Channel %d is not in dataOut.channelList")
|
|
|
channelIndexList.append(dataOut.channelList.index(channel))
|
|
|
|
|
|
if normFactor is None:
|
|
|
factor = dataOut.normFactor
|
|
|
else:
|
|
|
factor = normFactor
|
|
|
|
|
|
#factor = dataOut.normFactor
|
|
|
x = dataOut.getTimeRange()
|
|
|
y = dataOut.getHeiRange()
|
|
|
|
|
|
z = dataOut.data_spc/factor
|
|
|
z = numpy.where(numpy.isfinite(z), z, numpy.NAN)
|
|
|
avg = numpy.average(z, axis=1)
|
|
|
avgdB = 10.*numpy.log10(avg)
|
|
|
# avgdB = dataOut.getPower()
|
|
|
|
|
|
|
|
|
thisDatetime = dataOut.datatime
|
|
|
#thisDatetime = datetime.datetime.utcfromtimestamp(dataOut.getTimeRange()[0])
|
|
|
title = wintitle + " RTI" #: %s" %(thisDatetime.strftime("%d-%b-%Y"))
|
|
|
xlabel = ""
|
|
|
ylabel = "Range (Km)"
|
|
|
|
|
|
update_figfile = False
|
|
|
|
|
|
if self.xmax is not None and dataOut.ltctime >= self.xmax: #yong
|
|
|
self.counter_imagwr = wr_period
|
|
|
self.isConfig = False
|
|
|
update_figfile = True
|
|
|
|
|
|
if not self.isConfig:
|
|
|
|
|
|
nplots = len(channelIndexList)
|
|
|
|
|
|
self.setup(id=id,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
showprofile=showprofile,
|
|
|
show=show)
|
|
|
|
|
|
if timerange != None:
|
|
|
self.timerange = timerange
|
|
|
|
|
|
self.xmin, self.xmax = self.getTimeLim(x, xmin, xmax, timerange)
|
|
|
|
|
|
noise = dataOut.noise/factor
|
|
|
noisedB = 10*numpy.log10(noise)
|
|
|
|
|
|
if ymin == None: ymin = numpy.nanmin(y)
|
|
|
if ymax == None: ymax = numpy.nanmax(y)
|
|
|
if zmin == None: zmin = numpy.floor(numpy.nanmin(noisedB)) - 3
|
|
|
if zmax == None: zmax = numpy.ceil(numpy.nanmax(avgdB)) + 3
|
|
|
|
|
|
self.FTP_WEI = ftp_wei
|
|
|
self.EXP_CODE = exp_code
|
|
|
self.SUB_EXP_CODE = sub_exp_code
|
|
|
self.PLOT_POS = plot_pos
|
|
|
|
|
|
self.name = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
self.isConfig = True
|
|
|
self.figfile = figfile
|
|
|
update_figfile = True
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
for i in range(self.nplots):
|
|
|
index = channelIndexList[i]
|
|
|
title = "Channel %d: %s" %(dataOut.channelList[index], thisDatetime.strftime("%Y/%m/%d %H:%M:%S"))
|
|
|
if ((dataOut.azimuth!=None) and (dataOut.zenith!=None)):
|
|
|
title = title + '_' + 'azimuth,zenith=%2.2f,%2.2f'%(dataOut.azimuth, dataOut.zenith)
|
|
|
axes = self.axesList[i*self.__nsubplots]
|
|
|
zdB = avgdB[index].reshape((1,-1))
|
|
|
axes.pcolorbuffer(x, y, zdB,
|
|
|
xmin=self.xmin, xmax=self.xmax, ymin=ymin, ymax=ymax, zmin=zmin, zmax=zmax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title, rti=True, XAxisAsTime=True,
|
|
|
ticksize=9, cblabel='', cbsize="1%", colormap=colormap)
|
|
|
|
|
|
if self.__showprofile:
|
|
|
axes = self.axesList[i*self.__nsubplots +1]
|
|
|
axes.pline(avgdB[index], y,
|
|
|
xmin=zmin, xmax=zmax, ymin=ymin, ymax=ymax,
|
|
|
xlabel='dB', ylabel='', title='',
|
|
|
ytick_visible=False,
|
|
|
grid='x')
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
self.save(figpath=figpath,
|
|
|
figfile=figfile,
|
|
|
save=save,
|
|
|
ftp=ftp,
|
|
|
wr_period=wr_period,
|
|
|
thisDatetime=thisDatetime,
|
|
|
update_figfile=update_figfile)
|
|
|
return dataOut
|
|
|
|
|
|
@MPDecorator
|
|
|
class CoherenceMap_(Figure):
|
|
|
isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
WIDTHPROF = None
|
|
|
HEIGHTPROF = None
|
|
|
PREFIX = 'cmap'
|
|
|
|
|
|
def __init__(self):
|
|
|
Figure.__init__(self)
|
|
|
self.timerange = 2*60*60
|
|
|
self.isConfig = False
|
|
|
self.__nsubplots = 1
|
|
|
|
|
|
self.WIDTH = 800
|
|
|
self.HEIGHT = 180
|
|
|
self.WIDTHPROF = 120
|
|
|
self.HEIGHTPROF = 0
|
|
|
self.counter_imagwr = 0
|
|
|
|
|
|
self.PLOT_CODE = COH_CODE
|
|
|
|
|
|
self.FTP_WEI = None
|
|
|
self.EXP_CODE = None
|
|
|
self.SUB_EXP_CODE = None
|
|
|
self.PLOT_POS = None
|
|
|
self.counter_imagwr = 0
|
|
|
|
|
|
self.xmin = None
|
|
|
self.xmax = None
|
|
|
|
|
|
def getSubplots(self):
|
|
|
ncol = 1
|
|
|
nrow = self.nplots*2
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, id, nplots, wintitle, showprofile=True, show=True):
|
|
|
self.__showprofile = showprofile
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 1
|
|
|
colspan = 1
|
|
|
if showprofile:
|
|
|
ncolspan = 7
|
|
|
colspan = 6
|
|
|
self.__nsubplots = 2
|
|
|
|
|
|
self.createFigure(id = id,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = self.WIDTH + self.WIDTHPROF,
|
|
|
heightplot = self.HEIGHT + self.HEIGHTPROF,
|
|
|
show=True)
|
|
|
|
|
|
nrow, ncol = self.getSubplots()
|
|
|
|
|
|
for y in range(nrow):
|
|
|
for x in range(ncol):
|
|
|
|
|
|
self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan, colspan, 1)
|
|
|
|
|
|
if showprofile:
|
|
|
self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan+colspan, 1, 1)
|
|
|
|
|
|
def run(self, dataOut, id, wintitle="", pairsList=None, showprofile='True',
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None,
|
|
|
timerange=None, phase_min=None, phase_max=None,
|
|
|
save=False, figpath='./', figfile=None, ftp=False, wr_period=1,
|
|
|
coherence_cmap='jet', phase_cmap='RdBu_r', show=True,
|
|
|
server=None, folder=None, username=None, password=None,
|
|
|
ftp_wei=0, exp_code=0, sub_exp_code=0, plot_pos=0):
|
|
|
|
|
|
|
|
|
if dataOut.flagNoData:
|
|
|
return dataOut
|
|
|
|
|
|
if not isTimeInHourRange(dataOut.datatime, xmin, xmax):
|
|
|
return
|
|
|
|
|
|
if pairsList == None:
|
|
|
pairsIndexList = dataOut.pairsIndexList
|
|
|
else:
|
|
|
pairsIndexList = []
|
|
|
for pair in pairsList:
|
|
|
if pair not in dataOut.pairsList:
|
|
|
raise ValueError("Pair %s is not in dataOut.pairsList" %(pair))
|
|
|
pairsIndexList.append(dataOut.pairsList.index(pair))
|
|
|
|
|
|
if pairsIndexList == []:
|
|
|
return
|
|
|
|
|
|
if len(pairsIndexList) > 4:
|
|
|
pairsIndexList = pairsIndexList[0:4]
|
|
|
|
|
|
if phase_min == None:
|
|
|
phase_min = -180
|
|
|
if phase_max == None:
|
|
|
phase_max = 180
|
|
|
|
|
|
x = dataOut.getTimeRange()
|
|
|
y = dataOut.getHeiRange()
|
|
|
|
|
|
thisDatetime = dataOut.datatime
|
|
|
|
|
|
title = wintitle + " CoherenceMap" #: %s" %(thisDatetime.strftime("%d-%b-%Y"))
|
|
|
xlabel = ""
|
|
|
ylabel = "Range (Km)"
|
|
|
update_figfile = False
|
|
|
|
|
|
if not self.isConfig:
|
|
|
nplots = len(pairsIndexList)
|
|
|
self.setup(id=id,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
showprofile=showprofile,
|
|
|
show=show)
|
|
|
|
|
|
if timerange != None:
|
|
|
self.timerange = timerange
|
|
|
|
|
|
self.xmin, self.xmax = self.getTimeLim(x, xmin, xmax, timerange)
|
|
|
|
|
|
if ymin == None: ymin = numpy.nanmin(y)
|
|
|
if ymax == None: ymax = numpy.nanmax(y)
|
|
|
if zmin == None: zmin = 0.
|
|
|
if zmax == None: zmax = 1.
|
|
|
|
|
|
self.FTP_WEI = ftp_wei
|
|
|
self.EXP_CODE = exp_code
|
|
|
self.SUB_EXP_CODE = sub_exp_code
|
|
|
self.PLOT_POS = plot_pos
|
|
|
|
|
|
self.name = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
|
|
|
self.isConfig = True
|
|
|
update_figfile = True
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
for i in range(self.nplots):
|
|
|
|
|
|
pair = dataOut.pairsList[pairsIndexList[i]]
|
|
|
|
|
|
ccf = numpy.average(dataOut.data_cspc[pairsIndexList[i],:,:],axis=0)
|
|
|
powa = numpy.average(dataOut.data_spc[pair[0],:,:],axis=0)
|
|
|
powb = numpy.average(dataOut.data_spc[pair[1],:,:],axis=0)
|
|
|
|
|
|
|
|
|
avgcoherenceComplex = ccf/numpy.sqrt(powa*powb)
|
|
|
coherence = numpy.abs(avgcoherenceComplex)
|
|
|
|
|
|
z = coherence.reshape((1,-1))
|
|
|
|
|
|
counter = 0
|
|
|
|
|
|
title = "Coherence Ch%d * Ch%d: %s" %(pair[0], pair[1], thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
axes = self.axesList[i*self.__nsubplots*2]
|
|
|
axes.pcolorbuffer(x, y, z,
|
|
|
xmin=self.xmin, xmax=self.xmax, ymin=ymin, ymax=ymax, zmin=zmin, zmax=zmax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title, rti=True, XAxisAsTime=True,
|
|
|
ticksize=9, cblabel='', colormap=coherence_cmap, cbsize="1%")
|
|
|
|
|
|
if self.__showprofile:
|
|
|
counter += 1
|
|
|
axes = self.axesList[i*self.__nsubplots*2 + counter]
|
|
|
axes.pline(coherence, y,
|
|
|
xmin=zmin, xmax=zmax, ymin=ymin, ymax=ymax,
|
|
|
xlabel='', ylabel='', title='', ticksize=7,
|
|
|
ytick_visible=False, nxticks=5,
|
|
|
grid='x')
|
|
|
|
|
|
counter += 1
|
|
|
|
|
|
phase = numpy.arctan2(avgcoherenceComplex.imag, avgcoherenceComplex.real)*180/numpy.pi
|
|
|
|
|
|
z = phase.reshape((1,-1))
|
|
|
|
|
|
title = "Phase Ch%d * Ch%d: %s" %(pair[0], pair[1], thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
axes = self.axesList[i*self.__nsubplots*2 + counter]
|
|
|
axes.pcolorbuffer(x, y, z,
|
|
|
xmin=self.xmin, xmax=self.xmax, ymin=ymin, ymax=ymax, zmin=phase_min, zmax=phase_max,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title, rti=True, XAxisAsTime=True,
|
|
|
ticksize=9, cblabel='', colormap=phase_cmap, cbsize="1%")
|
|
|
|
|
|
if self.__showprofile:
|
|
|
counter += 1
|
|
|
axes = self.axesList[i*self.__nsubplots*2 + counter]
|
|
|
axes.pline(phase, y,
|
|
|
xmin=phase_min, xmax=phase_max, ymin=ymin, ymax=ymax,
|
|
|
xlabel='', ylabel='', title='', ticksize=7,
|
|
|
ytick_visible=False, nxticks=4,
|
|
|
grid='x')
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
if dataOut.ltctime >= self.xmax:
|
|
|
self.counter_imagwr = wr_period
|
|
|
self.isConfig = False
|
|
|
update_figfile = True
|
|
|
|
|
|
self.save(figpath=figpath,
|
|
|
figfile=figfile,
|
|
|
save=save,
|
|
|
ftp=ftp,
|
|
|
wr_period=wr_period,
|
|
|
thisDatetime=thisDatetime,
|
|
|
update_figfile=update_figfile)
|
|
|
|
|
|
return dataOut
|
|
|
|
|
|
@MPDecorator
|
|
|
class PowerProfilePlot_(Figure):
|
|
|
|
|
|
isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
WIDTHPROF = None
|
|
|
HEIGHTPROF = None
|
|
|
PREFIX = 'spcprofile'
|
|
|
|
|
|
def __init__(self):
|
|
|
Figure.__init__(self)
|
|
|
self.isConfig = False
|
|
|
self.__nsubplots = 1
|
|
|
|
|
|
self.PLOT_CODE = POWER_CODE
|
|
|
|
|
|
self.WIDTH = 300
|
|
|
self.HEIGHT = 500
|
|
|
self.counter_imagwr = 0
|
|
|
|
|
|
def getSubplots(self):
|
|
|
ncol = 1
|
|
|
nrow = 1
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, id, nplots, wintitle, show):
|
|
|
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 1
|
|
|
colspan = 1
|
|
|
|
|
|
self.createFigure(id = id,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = self.WIDTH,
|
|
|
heightplot = self.HEIGHT,
|
|
|
show=show)
|
|
|
|
|
|
nrow, ncol = self.getSubplots()
|
|
|
|
|
|
counter = 0
|
|
|
for y in range(nrow):
|
|
|
for x in range(ncol):
|
|
|
self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan, colspan, 1)
|
|
|
|
|
|
def run(self, dataOut, id, wintitle="", channelList=None,
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None,
|
|
|
save=False, figpath='./', figfile=None, show=True,
|
|
|
ftp=False, wr_period=1, server=None,
|
|
|
folder=None, username=None, password=None):
|
|
|
|
|
|
if dataOut.flagNoData:
|
|
|
return dataOut
|
|
|
|
|
|
|
|
|
if channelList == None:
|
|
|
channelIndexList = dataOut.channelIndexList
|
|
|
channelList = dataOut.channelList
|
|
|
else:
|
|
|
channelIndexList = []
|
|
|
for channel in channelList:
|
|
|
if channel not in dataOut.channelList:
|
|
|
raise ValueError("Channel %d is not in dataOut.channelList")
|
|
|
channelIndexList.append(dataOut.channelList.index(channel))
|
|
|
|
|
|
factor = dataOut.normFactor
|
|
|
|
|
|
y = dataOut.getHeiRange()
|
|
|
|
|
|
#for voltage
|
|
|
if dataOut.type == 'Voltage':
|
|
|
x = dataOut.data[channelIndexList,:] * numpy.conjugate(dataOut.data[channelIndexList,:])
|
|
|
x = x.real
|
|
|
x = numpy.where(numpy.isfinite(x), x, numpy.NAN)
|
|
|
|
|
|
#for spectra
|
|
|
if dataOut.type == 'Spectra':
|
|
|
x = dataOut.data_spc[channelIndexList,:,:]/factor
|
|
|
x = numpy.where(numpy.isfinite(x), x, numpy.NAN)
|
|
|
x = numpy.average(x, axis=1)
|
|
|
|
|
|
|
|
|
xdB = 10*numpy.log10(x)
|
|
|
|
|
|
thisDatetime = datetime.datetime.utcfromtimestamp(dataOut.getTimeRange()[0])
|
|
|
title = wintitle + " Power Profile %s" %(thisDatetime.strftime("%d-%b-%Y"))
|
|
|
xlabel = "dB"
|
|
|
ylabel = "Range (Km)"
|
|
|
|
|
|
if not self.isConfig:
|
|
|
|
|
|
nplots = 1
|
|
|
|
|
|
self.setup(id=id,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
show=show)
|
|
|
|
|
|
if ymin == None: ymin = numpy.nanmin(y)
|
|
|
if ymax == None: ymax = numpy.nanmax(y)
|
|
|
if xmin == None: xmin = numpy.nanmin(xdB)*0.9
|
|
|
if xmax == None: xmax = numpy.nanmax(xdB)*1.1
|
|
|
|
|
|
self.isConfig = True
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
title = "Power Profile: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
axes = self.axesList[0]
|
|
|
|
|
|
legendlabels = ["channel %d"%x for x in channelList]
|
|
|
axes.pmultiline(xdB, y,
|
|
|
xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title, legendlabels=legendlabels,
|
|
|
ytick_visible=True, nxticks=5,
|
|
|
grid='x')
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
self.save(figpath=figpath,
|
|
|
figfile=figfile,
|
|
|
save=save,
|
|
|
ftp=ftp,
|
|
|
wr_period=wr_period,
|
|
|
thisDatetime=thisDatetime)
|
|
|
|
|
|
return dataOut
|
|
|
|
|
|
@MPDecorator
|
|
|
class SpectraCutPlot_(Figure):
|
|
|
|
|
|
isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
WIDTHPROF = None
|
|
|
HEIGHTPROF = None
|
|
|
PREFIX = 'spc_cut'
|
|
|
|
|
|
def __init__(self):
|
|
|
Figure.__init__(self)
|
|
|
self.isConfig = False
|
|
|
self.__nsubplots = 1
|
|
|
|
|
|
self.PLOT_CODE = POWER_CODE
|
|
|
|
|
|
self.WIDTH = 700
|
|
|
self.HEIGHT = 500
|
|
|
self.counter_imagwr = 0
|
|
|
|
|
|
def getSubplots(self):
|
|
|
ncol = 1
|
|
|
nrow = 1
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, id, nplots, wintitle, show):
|
|
|
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 1
|
|
|
colspan = 1
|
|
|
|
|
|
self.createFigure(id = id,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = self.WIDTH,
|
|
|
heightplot = self.HEIGHT,
|
|
|
show=show)
|
|
|
|
|
|
nrow, ncol = self.getSubplots()
|
|
|
|
|
|
counter = 0
|
|
|
for y in range(nrow):
|
|
|
for x in range(ncol):
|
|
|
self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan, colspan, 1)
|
|
|
|
|
|
def run(self, dataOut, id, wintitle="", channelList=None,
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None,
|
|
|
save=False, figpath='./', figfile=None, show=True,
|
|
|
ftp=False, wr_period=1, server=None,
|
|
|
folder=None, username=None, password=None,
|
|
|
xaxis="frequency"):
|
|
|
|
|
|
if dataOut.flagNoData:
|
|
|
return dataOut
|
|
|
|
|
|
if channelList == None:
|
|
|
channelIndexList = dataOut.channelIndexList
|
|
|
channelList = dataOut.channelList
|
|
|
else:
|
|
|
channelIndexList = []
|
|
|
for channel in channelList:
|
|
|
if channel not in dataOut.channelList:
|
|
|
raise ValueError("Channel %d is not in dataOut.channelList")
|
|
|
channelIndexList.append(dataOut.channelList.index(channel))
|
|
|
|
|
|
factor = dataOut.normFactor
|
|
|
|
|
|
y = dataOut.getHeiRange()
|
|
|
|
|
|
z = dataOut.data_spc/factor
|
|
|
z = numpy.where(numpy.isfinite(z), z, numpy.NAN)
|
|
|
|
|
|
hei_index = numpy.arange(25)*3 + 20
|
|
|
|
|
|
if xaxis == "frequency":
|
|
|
x = dataOut.getFreqRange()/1000.
|
|
|
zdB = 10*numpy.log10(z[0,:,hei_index])
|
|
|
xlabel = "Frequency (kHz)"
|
|
|
ylabel = "Power (dB)"
|
|
|
|
|
|
elif xaxis == "time":
|
|
|
x = dataOut.getAcfRange()
|
|
|
zdB = z[0,:,hei_index]
|
|
|
xlabel = "Time (ms)"
|
|
|
ylabel = "ACF"
|
|
|
|
|
|
else:
|
|
|
x = dataOut.getVelRange()
|
|
|
zdB = 10*numpy.log10(z[0,:,hei_index])
|
|
|
xlabel = "Velocity (m/s)"
|
|
|
ylabel = "Power (dB)"
|
|
|
|
|
|
thisDatetime = datetime.datetime.utcfromtimestamp(dataOut.getTimeRange()[0])
|
|
|
title = wintitle + " Range Cuts %s" %(thisDatetime.strftime("%d-%b-%Y"))
|
|
|
|
|
|
if not self.isConfig:
|
|
|
|
|
|
nplots = 1
|
|
|
|
|
|
self.setup(id=id,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
show=show)
|
|
|
|
|
|
if xmin == None: xmin = numpy.nanmin(x)*0.9
|
|
|
if xmax == None: xmax = numpy.nanmax(x)*1.1
|
|
|
if ymin == None: ymin = numpy.nanmin(zdB)
|
|
|
if ymax == None: ymax = numpy.nanmax(zdB)
|
|
|
|
|
|
self.isConfig = True
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
title = "Spectra Cuts: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
axes = self.axesList[0]
|
|
|
|
|
|
legendlabels = ["Range = %dKm" %y[i] for i in hei_index]
|
|
|
|
|
|
axes.pmultilineyaxis( x, zdB,
|
|
|
xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title, legendlabels=legendlabels,
|
|
|
ytick_visible=True, nxticks=5,
|
|
|
grid='x')
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
self.save(figpath=figpath,
|
|
|
figfile=figfile,
|
|
|
save=save,
|
|
|
ftp=ftp,
|
|
|
wr_period=wr_period,
|
|
|
thisDatetime=thisDatetime)
|
|
|
|
|
|
return dataOut
|
|
|
|
|
|
@MPDecorator
|
|
|
class Noise_(Figure):
|
|
|
|
|
|
isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
PREFIX = 'noise'
|
|
|
|
|
|
|
|
|
def __init__(self):
|
|
|
Figure.__init__(self)
|
|
|
self.timerange = 24*60*60
|
|
|
self.isConfig = False
|
|
|
self.__nsubplots = 1
|
|
|
self.counter_imagwr = 0
|
|
|
self.WIDTH = 800
|
|
|
self.HEIGHT = 400
|
|
|
self.WIDTHPROF = 120
|
|
|
self.HEIGHTPROF = 0
|
|
|
self.xdata = None
|
|
|
self.ydata = None
|
|
|
|
|
|
self.PLOT_CODE = NOISE_CODE
|
|
|
|
|
|
self.FTP_WEI = None
|
|
|
self.EXP_CODE = None
|
|
|
self.SUB_EXP_CODE = None
|
|
|
self.PLOT_POS = None
|
|
|
self.figfile = None
|
|
|
|
|
|
self.xmin = None
|
|
|
self.xmax = None
|
|
|
|
|
|
def getSubplots(self):
|
|
|
|
|
|
ncol = 1
|
|
|
nrow = 1
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def openfile(self, filename):
|
|
|
dirname = os.path.dirname(filename)
|
|
|
|
|
|
if not os.path.exists(dirname):
|
|
|
os.mkdir(dirname)
|
|
|
|
|
|
f = open(filename,'w+')
|
|
|
f.write('\n\n')
|
|
|
f.write('JICAMARCA RADIO OBSERVATORY - Noise \n')
|
|
|
f.write('DD MM YYYY HH MM SS Channel0 Channel1 Channel2 Channel3\n\n' )
|
|
|
f.close()
|
|
|
|
|
|
def save_data(self, filename_phase, data, data_datetime):
|
|
|
|
|
|
f=open(filename_phase,'a')
|
|
|
|
|
|
timetuple_data = data_datetime.timetuple()
|
|
|
day = str(timetuple_data.tm_mday)
|
|
|
month = str(timetuple_data.tm_mon)
|
|
|
year = str(timetuple_data.tm_year)
|
|
|
hour = str(timetuple_data.tm_hour)
|
|
|
minute = str(timetuple_data.tm_min)
|
|
|
second = str(timetuple_data.tm_sec)
|
|
|
|
|
|
data_msg = ''
|
|
|
for i in range(len(data)):
|
|
|
data_msg += str(data[i]) + ' '
|
|
|
|
|
|
f.write(day+' '+month+' '+year+' '+hour+' '+minute+' '+second+' ' + data_msg + '\n')
|
|
|
f.close()
|
|
|
|
|
|
|
|
|
def setup(self, id, nplots, wintitle, showprofile=True, show=True):
|
|
|
|
|
|
self.__showprofile = showprofile
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 7
|
|
|
colspan = 6
|
|
|
self.__nsubplots = 2
|
|
|
|
|
|
self.createFigure(id = id,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = self.WIDTH+self.WIDTHPROF,
|
|
|
heightplot = self.HEIGHT+self.HEIGHTPROF,
|
|
|
show=show)
|
|
|
|
|
|
nrow, ncol = self.getSubplots()
|
|
|
|
|
|
self.addAxes(nrow, ncol*ncolspan, 0, 0, colspan, 1)
|
|
|
|
|
|
|
|
|
def run(self, dataOut, id, wintitle="", channelList=None, showprofile='True',
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None,
|
|
|
timerange=None,
|
|
|
save=False, figpath='./', figfile=None, show=True, ftp=False, wr_period=1,
|
|
|
server=None, folder=None, username=None, password=None,
|
|
|
ftp_wei=0, exp_code=0, sub_exp_code=0, plot_pos=0):
|
|
|
|
|
|
if dataOut.flagNoData:
|
|
|
return dataOut
|
|
|
|
|
|
if not isTimeInHourRange(dataOut.datatime, xmin, xmax):
|
|
|
return
|
|
|
|
|
|
if channelList == None:
|
|
|
channelIndexList = dataOut.channelIndexList
|
|
|
channelList = dataOut.channelList
|
|
|
else:
|
|
|
channelIndexList = []
|
|
|
for channel in channelList:
|
|
|
if channel not in dataOut.channelList:
|
|
|
raise ValueError("Channel %d is not in dataOut.channelList")
|
|
|
channelIndexList.append(dataOut.channelList.index(channel))
|
|
|
|
|
|
x = dataOut.getTimeRange()
|
|
|
#y = dataOut.getHeiRange()
|
|
|
factor = dataOut.normFactor
|
|
|
noise = dataOut.noise[channelIndexList]/factor
|
|
|
noisedB = 10*numpy.log10(noise)
|
|
|
|
|
|
thisDatetime = dataOut.datatime
|
|
|
|
|
|
title = wintitle + " Noise" # : %s" %(thisDatetime.strftime("%d-%b-%Y"))
|
|
|
xlabel = ""
|
|
|
ylabel = "Intensity (dB)"
|
|
|
update_figfile = False
|
|
|
|
|
|
if not self.isConfig:
|
|
|
|
|
|
nplots = 1
|
|
|
|
|
|
self.setup(id=id,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
showprofile=showprofile,
|
|
|
show=show)
|
|
|
|
|
|
if timerange != None:
|
|
|
self.timerange = timerange
|
|
|
|
|
|
self.xmin, self.xmax = self.getTimeLim(x, xmin, xmax, timerange)
|
|
|
|
|
|
if ymin == None: ymin = numpy.floor(numpy.nanmin(noisedB)) - 10.0
|
|
|
if ymax == None: ymax = numpy.nanmax(noisedB) + 10.0
|
|
|
|
|
|
self.FTP_WEI = ftp_wei
|
|
|
self.EXP_CODE = exp_code
|
|
|
self.SUB_EXP_CODE = sub_exp_code
|
|
|
self.PLOT_POS = plot_pos
|
|
|
|
|
|
|
|
|
self.name = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
self.isConfig = True
|
|
|
self.figfile = figfile
|
|
|
self.xdata = numpy.array([])
|
|
|
self.ydata = numpy.array([])
|
|
|
|
|
|
update_figfile = True
|
|
|
|
|
|
#open file beacon phase
|
|
|
path = '%s%03d' %(self.PREFIX, self.id)
|
|
|
noise_file = os.path.join(path,'%s.txt'%self.name)
|
|
|
self.filename_noise = os.path.join(figpath,noise_file)
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
title = "Noise %s" %(thisDatetime.strftime("%Y/%m/%d %H:%M:%S"))
|
|
|
|
|
|
legendlabels = ["channel %d"%(idchannel) for idchannel in channelList]
|
|
|
axes = self.axesList[0]
|
|
|
|
|
|
self.xdata = numpy.hstack((self.xdata, x[0:1]))
|
|
|
|
|
|
if len(self.ydata)==0:
|
|
|
self.ydata = noisedB.reshape(-1,1)
|
|
|
else:
|
|
|
self.ydata = numpy.hstack((self.ydata, noisedB.reshape(-1,1)))
|
|
|
|
|
|
|
|
|
axes.pmultilineyaxis(x=self.xdata, y=self.ydata,
|
|
|
xmin=self.xmin, xmax=self.xmax, ymin=ymin, ymax=ymax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title, legendlabels=legendlabels, marker='x', markersize=8, linestyle="solid",
|
|
|
XAxisAsTime=True, grid='both'
|
|
|
)
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
if dataOut.ltctime >= self.xmax:
|
|
|
self.counter_imagwr = wr_period
|
|
|
self.isConfig = False
|
|
|
update_figfile = True
|
|
|
|
|
|
self.save(figpath=figpath,
|
|
|
figfile=figfile,
|
|
|
save=save,
|
|
|
ftp=ftp,
|
|
|
wr_period=wr_period,
|
|
|
thisDatetime=thisDatetime,
|
|
|
update_figfile=update_figfile)
|
|
|
|
|
|
#store data beacon phase
|
|
|
if save:
|
|
|
self.save_data(self.filename_noise, noisedB, thisDatetime)
|
|
|
|
|
|
return dataOut
|
|
|
|
|
|
@MPDecorator
|
|
|
class BeaconPhase_(Figure):
|
|
|
|
|
|
__isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
PREFIX = 'beacon_phase'
|
|
|
|
|
|
def __init__(self):
|
|
|
Figure.__init__(self)
|
|
|
self.timerange = 24*60*60
|
|
|
self.isConfig = False
|
|
|
self.__nsubplots = 1
|
|
|
self.counter_imagwr = 0
|
|
|
self.WIDTH = 800
|
|
|
self.HEIGHT = 400
|
|
|
self.WIDTHPROF = 120
|
|
|
self.HEIGHTPROF = 0
|
|
|
self.xdata = None
|
|
|
self.ydata = None
|
|
|
|
|
|
self.PLOT_CODE = BEACON_CODE
|
|
|
|
|
|
self.FTP_WEI = None
|
|
|
self.EXP_CODE = None
|
|
|
self.SUB_EXP_CODE = None
|
|
|
self.PLOT_POS = None
|
|
|
|
|
|
self.filename_phase = None
|
|
|
|
|
|
self.figfile = None
|
|
|
|
|
|
self.xmin = None
|
|
|
self.xmax = None
|
|
|
|
|
|
def getSubplots(self):
|
|
|
|
|
|
ncol = 1
|
|
|
nrow = 1
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, id, nplots, wintitle, showprofile=True, show=True):
|
|
|
|
|
|
self.__showprofile = showprofile
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 7
|
|
|
colspan = 6
|
|
|
self.__nsubplots = 2
|
|
|
|
|
|
self.createFigure(id = id,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = self.WIDTH+self.WIDTHPROF,
|
|
|
heightplot = self.HEIGHT+self.HEIGHTPROF,
|
|
|
show=show)
|
|
|
|
|
|
nrow, ncol = self.getSubplots()
|
|
|
|
|
|
self.addAxes(nrow, ncol*ncolspan, 0, 0, colspan, 1)
|
|
|
|
|
|
def save_phase(self, filename_phase):
|
|
|
f = open(filename_phase,'w+')
|
|
|
f.write('\n\n')
|
|
|
f.write('JICAMARCA RADIO OBSERVATORY - Beacon Phase \n')
|
|
|
f.write('DD MM YYYY HH MM SS pair(2,0) pair(2,1) pair(2,3) pair(2,4)\n\n' )
|
|
|
f.close()
|
|
|
|
|
|
def save_data(self, filename_phase, data, data_datetime):
|
|
|
f=open(filename_phase,'a')
|
|
|
timetuple_data = data_datetime.timetuple()
|
|
|
day = str(timetuple_data.tm_mday)
|
|
|
month = str(timetuple_data.tm_mon)
|
|
|
year = str(timetuple_data.tm_year)
|
|
|
hour = str(timetuple_data.tm_hour)
|
|
|
minute = str(timetuple_data.tm_min)
|
|
|
second = str(timetuple_data.tm_sec)
|
|
|
f.write(day+' '+month+' '+year+' '+hour+' '+minute+' '+second+' '+str(data[0])+' '+str(data[1])+' '+str(data[2])+' '+str(data[3])+'\n')
|
|
|
f.close()
|
|
|
|
|
|
|
|
|
def run(self, dataOut, id, wintitle="", pairsList=None, showprofile='True',
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, hmin=None, hmax=None,
|
|
|
timerange=None,
|
|
|
save=False, figpath='./', figfile=None, show=True, ftp=False, wr_period=1,
|
|
|
server=None, folder=None, username=None, password=None,
|
|
|
ftp_wei=0, exp_code=0, sub_exp_code=0, plot_pos=0):
|
|
|
|
|
|
if dataOut.flagNoData:
|
|
|
return dataOut
|
|
|
|
|
|
if not isTimeInHourRange(dataOut.datatime, xmin, xmax):
|
|
|
return
|
|
|
|
|
|
if pairsList == None:
|
|
|
pairsIndexList = dataOut.pairsIndexList[:10]
|
|
|
else:
|
|
|
pairsIndexList = []
|
|
|
for pair in pairsList:
|
|
|
if pair not in dataOut.pairsList:
|
|
|
raise ValueError("Pair %s is not in dataOut.pairsList" %(pair))
|
|
|
pairsIndexList.append(dataOut.pairsList.index(pair))
|
|
|
|
|
|
if pairsIndexList == []:
|
|
|
return
|
|
|
|
|
|
# if len(pairsIndexList) > 4:
|
|
|
# pairsIndexList = pairsIndexList[0:4]
|
|
|
|
|
|
hmin_index = None
|
|
|
hmax_index = None
|
|
|
|
|
|
if hmin != None and hmax != None:
|
|
|
indexes = numpy.arange(dataOut.nHeights)
|
|
|
hmin_list = indexes[dataOut.heightList >= hmin]
|
|
|
hmax_list = indexes[dataOut.heightList <= hmax]
|
|
|
|
|
|
if hmin_list.any():
|
|
|
hmin_index = hmin_list[0]
|
|
|
|
|
|
if hmax_list.any():
|
|
|
hmax_index = hmax_list[-1]+1
|
|
|
|
|
|
x = dataOut.getTimeRange()
|
|
|
#y = dataOut.getHeiRange()
|
|
|
|
|
|
|
|
|
thisDatetime = dataOut.datatime
|
|
|
|
|
|
title = wintitle + " Signal Phase" # : %s" %(thisDatetime.strftime("%d-%b-%Y"))
|
|
|
xlabel = "Local Time"
|
|
|
ylabel = "Phase (degrees)"
|
|
|
|
|
|
update_figfile = False
|
|
|
|
|
|
nplots = len(pairsIndexList)
|
|
|
#phase = numpy.zeros((len(pairsIndexList),len(dataOut.beacon_heiIndexList)))
|
|
|
phase_beacon = numpy.zeros(len(pairsIndexList))
|
|
|
for i in range(nplots):
|
|
|
pair = dataOut.pairsList[pairsIndexList[i]]
|
|
|
ccf = numpy.average(dataOut.data_cspc[pairsIndexList[i], :, hmin_index:hmax_index], axis=0)
|
|
|
powa = numpy.average(dataOut.data_spc[pair[0], :, hmin_index:hmax_index], axis=0)
|
|
|
powb = numpy.average(dataOut.data_spc[pair[1], :, hmin_index:hmax_index], axis=0)
|
|
|
avgcoherenceComplex = ccf/numpy.sqrt(powa*powb)
|
|
|
phase = numpy.arctan2(avgcoherenceComplex.imag, avgcoherenceComplex.real)*180/numpy.pi
|
|
|
|
|
|
if dataOut.beacon_heiIndexList:
|
|
|
phase_beacon[i] = numpy.average(phase[dataOut.beacon_heiIndexList])
|
|
|
else:
|
|
|
phase_beacon[i] = numpy.average(phase)
|
|
|
|
|
|
if not self.isConfig:
|
|
|
|
|
|
nplots = len(pairsIndexList)
|
|
|
|
|
|
self.setup(id=id,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
showprofile=showprofile,
|
|
|
show=show)
|
|
|
|
|
|
if timerange != None:
|
|
|
self.timerange = timerange
|
|
|
|
|
|
self.xmin, self.xmax = self.getTimeLim(x, xmin, xmax, timerange)
|
|
|
|
|
|
if ymin == None: ymin = 0
|
|
|
if ymax == None: ymax = 360
|
|
|
|
|
|
self.FTP_WEI = ftp_wei
|
|
|
self.EXP_CODE = exp_code
|
|
|
self.SUB_EXP_CODE = sub_exp_code
|
|
|
self.PLOT_POS = plot_pos
|
|
|
|
|
|
self.name = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
self.isConfig = True
|
|
|
self.figfile = figfile
|
|
|
self.xdata = numpy.array([])
|
|
|
self.ydata = numpy.array([])
|
|
|
|
|
|
update_figfile = True
|
|
|
|
|
|
#open file beacon phase
|
|
|
path = '%s%03d' %(self.PREFIX, self.id)
|
|
|
beacon_file = os.path.join(path,'%s.txt'%self.name)
|
|
|
self.filename_phase = os.path.join(figpath,beacon_file)
|
|
|
#self.save_phase(self.filename_phase)
|
|
|
|
|
|
|
|
|
#store data beacon phase
|
|
|
#self.save_data(self.filename_phase, phase_beacon, thisDatetime)
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
|
|
|
title = "Phase Plot %s" %(thisDatetime.strftime("%Y/%m/%d %H:%M:%S"))
|
|
|
|
|
|
legendlabels = ["Pair (%d,%d)"%(pair[0], pair[1]) for pair in dataOut.pairsList]
|
|
|
|
|
|
axes = self.axesList[0]
|
|
|
|
|
|
self.xdata = numpy.hstack((self.xdata, x[0:1]))
|
|
|
|
|
|
if len(self.ydata)==0:
|
|
|
self.ydata = phase_beacon.reshape(-1,1)
|
|
|
else:
|
|
|
self.ydata = numpy.hstack((self.ydata, phase_beacon.reshape(-1,1)))
|
|
|
|
|
|
|
|
|
axes.pmultilineyaxis(x=self.xdata, y=self.ydata,
|
|
|
xmin=self.xmin, xmax=self.xmax, ymin=ymin, ymax=ymax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title, legendlabels=legendlabels, marker='x', markersize=8, linestyle="solid",
|
|
|
XAxisAsTime=True, grid='both'
|
|
|
)
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
if dataOut.ltctime >= self.xmax:
|
|
|
self.counter_imagwr = wr_period
|
|
|
self.isConfig = False
|
|
|
update_figfile = True
|
|
|
|
|
|
self.save(figpath=figpath,
|
|
|
figfile=figfile,
|
|
|
save=save,
|
|
|
ftp=ftp,
|
|
|
wr_period=wr_period,
|
|
|
thisDatetime=thisDatetime,
|
|
|
update_figfile=update_figfile)
|
|
|
|
|
|
return dataOut
|
