|
|
import numpy
|
|
|
import time, datetime
|
|
|
from graphics.figure import *
|
|
|
|
|
|
class CrossSpectraPlot(Figure):
|
|
|
|
|
|
__isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
WIDTH = None
|
|
|
HEIGHT = None
|
|
|
WIDTHPROF = None
|
|
|
HEIGHTPROF = None
|
|
|
PREFIX = 'cspc'
|
|
|
|
|
|
def __init__(self):
|
|
|
|
|
|
self.__isConfig = False
|
|
|
self.__nsubplots = 4
|
|
|
|
|
|
self.WIDTH = 250
|
|
|
self.HEIGHT = 250
|
|
|
self.WIDTHPROF = 0
|
|
|
self.HEIGHTPROF = 0
|
|
|
|
|
|
def getSubplots(self):
|
|
|
|
|
|
ncol = 4
|
|
|
nrow = self.nplots
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, idfigure, nplots, wintitle, showprofile=True):
|
|
|
|
|
|
self.__showprofile = showprofile
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 1
|
|
|
colspan = 1
|
|
|
|
|
|
self.createFigure(idfigure = idfigure,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = self.WIDTH + self.WIDTHPROF,
|
|
|
heightplot = self.HEIGHT + self.HEIGHTPROF)
|
|
|
|
|
|
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, idfigure, wintitle="", pairsList=None, showprofile='True',
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None, normalize=True,
|
|
|
save=False, figpath='./', figfile=None,
|
|
|
power_cmap='jet', coherence_cmap='jet', phase_cmap='RdBu_r'):
|
|
|
|
|
|
"""
|
|
|
|
|
|
Input:
|
|
|
dataOut :
|
|
|
idfigure :
|
|
|
wintitle :
|
|
|
channelList :
|
|
|
showProfile :
|
|
|
xmin : None,
|
|
|
xmax : None,
|
|
|
ymin : None,
|
|
|
ymax : None,
|
|
|
zmin : None,
|
|
|
zmax : None
|
|
|
"""
|
|
|
|
|
|
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]
|
|
|
|
|
|
factor = 1
|
|
|
if normalize:
|
|
|
factor = dataOut.normFactor
|
|
|
x = dataOut.getVelRange(1)
|
|
|
y = dataOut.getHeiRange()
|
|
|
z = dataOut.data_spc[:,:,:]/factor
|
|
|
|
|
|
avg = numpy.average(z, axis=1)
|
|
|
noise = dataOut.getNoise()/factor
|
|
|
|
|
|
zdB = 10*numpy.log10(z)
|
|
|
avgdB = 10*numpy.log10(avg)
|
|
|
noisedB = 10*numpy.log10(noise)
|
|
|
|
|
|
|
|
|
thisDatetime = dataOut.datatime
|
|
|
title = "Cross-Spectra: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
xlabel = "Velocity (m/s)"
|
|
|
ylabel = "Range (Km)"
|
|
|
|
|
|
if not self.__isConfig:
|
|
|
|
|
|
nplots = len(pairsIndexList)
|
|
|
|
|
|
self.setup(idfigure=idfigure,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
showprofile=showprofile)
|
|
|
|
|
|
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.nanmin(avgdB)*0.9
|
|
|
if zmax == None: zmax = numpy.nanmax(avgdB)*0.9
|
|
|
|
|
|
self.__isConfig = True
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
for i in range(self.nplots):
|
|
|
pair = dataOut.pairsList[pairsIndexList[i]]
|
|
|
|
|
|
title = "Channel %d: %4.2fdB" %(pair[0], noisedB[pair[0]])
|
|
|
zdB = 10.*numpy.log10(dataOut.data_spc[pair[0],:,:]/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 = "Channel %d: %4.2fdB" %(pair[1], noisedB[pair[1]])
|
|
|
zdB = 10.*numpy.log10(dataOut.data_spc[pair[1],:,:]/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[pair[0],:,:]*dataOut.data_spc[pair[1],:,:])
|
|
|
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 %d%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=0, zmax=1,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title,
|
|
|
ticksize=9, colormap=coherence_cmap, cblabel='')
|
|
|
|
|
|
title = "Phase %d%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=-180, zmax=180,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title,
|
|
|
ticksize=9, colormap=phase_cmap, cblabel='')
|
|
|
|
|
|
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
if save:
|
|
|
date = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
if figfile == None:
|
|
|
figfile = self.getFilename(name = date)
|
|
|
|
|
|
self.saveFigure(figpath, figfile)
|
|
|
|
|
|
|
|
|
class RTIPlot(Figure):
|
|
|
|
|
|
__isConfig = None
|
|
|
__nsubplots = None
|
|
|
__missing = 1E30
|
|
|
WIDTHPROF = None
|
|
|
HEIGHTPROF = None
|
|
|
PREFIX = 'rti'
|
|
|
|
|
|
def __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.x_buffer = None
|
|
|
self.avgdB_buffer = None
|
|
|
|
|
|
def getSubplots(self):
|
|
|
|
|
|
ncol = 1
|
|
|
nrow = self.nplots
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, idfigure, nplots, wintitle, showprofile=True):
|
|
|
|
|
|
self.__showprofile = showprofile
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 1
|
|
|
colspan = 1
|
|
|
widthplot = self.WIDTH
|
|
|
heightplot = self.HEIGHT
|
|
|
if showprofile:
|
|
|
ncolspan = 7
|
|
|
colspan = 6
|
|
|
self.__nsubplots = 2
|
|
|
widthplot += self.WIDTHPROF
|
|
|
heightplot += self.HEIGHTPROF
|
|
|
|
|
|
self.createFigure(idfigure = idfigure,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = widthplot,
|
|
|
heightplot = heightplot)
|
|
|
|
|
|
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, idfigure, wintitle="", channelList=None, showprofile='True',
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None, normalize=True,
|
|
|
timerange=None,
|
|
|
save=False, figpath='./', figfile=None):
|
|
|
|
|
|
"""
|
|
|
|
|
|
Input:
|
|
|
dataOut :
|
|
|
idfigure :
|
|
|
wintitle :
|
|
|
channelList :
|
|
|
showProfile :
|
|
|
xmin : None,
|
|
|
xmax : None,
|
|
|
ymin : None,
|
|
|
ymax : None,
|
|
|
zmin : None,
|
|
|
zmax : None
|
|
|
"""
|
|
|
|
|
|
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 timerange != None:
|
|
|
self.timerange = timerange
|
|
|
|
|
|
tmin = None
|
|
|
tmax = None
|
|
|
factor = 1
|
|
|
if normalize:
|
|
|
factor = dataOut.normFactor
|
|
|
x = dataOut.getTimeRange()
|
|
|
y = dataOut.getHeiRange()
|
|
|
|
|
|
z = dataOut.data_spc[channelIndexList,:,:]/factor
|
|
|
z = numpy.where(numpy.isfinite(z), z, numpy.NAN)
|
|
|
avg = numpy.average(z, axis=1)
|
|
|
|
|
|
avgdB = 10.*numpy.log10(avg)
|
|
|
|
|
|
|
|
|
thisDatetime = dataOut.datatime
|
|
|
title = "RTI: %s" %(thisDatetime.strftime("%d-%b-%Y"))
|
|
|
xlabel = "Velocity (m/s)"
|
|
|
ylabel = "Range (Km)"
|
|
|
|
|
|
if not self.__isConfig:
|
|
|
|
|
|
nplots = len(channelIndexList)
|
|
|
|
|
|
self.setup(idfigure=idfigure,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
showprofile=showprofile)
|
|
|
|
|
|
tmin, tmax = self.getTimeLim(x, xmin, xmax)
|
|
|
if ymin == None: ymin = numpy.nanmin(y)
|
|
|
if ymax == None: ymax = numpy.nanmax(y)
|
|
|
if zmin == None: zmin = numpy.nanmin(avgdB)*0.9
|
|
|
if zmax == None: zmax = numpy.nanmax(avgdB)*0.9
|
|
|
|
|
|
self.name = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
self.x_buffer = numpy.array([])
|
|
|
self.avgdB_buffer = numpy.array([])
|
|
|
self.__isConfig = True
|
|
|
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
if len(self.avgdB_buffer)==0:
|
|
|
self.avgdB_buffer = avgdB
|
|
|
newxdim = 1
|
|
|
newydim = -1
|
|
|
else:
|
|
|
if x[0]>self.x_buffer[-1]:
|
|
|
gap = avgdB.copy()
|
|
|
gap[:] = self.__missing
|
|
|
self.avgdB_buffer = numpy.hstack((self.avgdB_buffer, gap))
|
|
|
|
|
|
self.avgdB_buffer = numpy.hstack((self.avgdB_buffer, avgdB))
|
|
|
newxdim = -1
|
|
|
newydim = len(y)
|
|
|
|
|
|
self.x_buffer = numpy.hstack((self.x_buffer, x))
|
|
|
|
|
|
self.avgdB_buffer = numpy.ma.masked_inside(self.avgdB_buffer,0.99*self.__missing,1.01*self.__missing)
|
|
|
|
|
|
for i in range(self.nplots):
|
|
|
title = "Channel %d: %s" %(dataOut.channelList[i], thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
axes = self.axesList[i*self.__nsubplots]
|
|
|
zdB = self.avgdB_buffer[i].reshape(newxdim,newydim)
|
|
|
axes.pcolor(self.x_buffer, y, zdB,
|
|
|
xmin=tmin, xmax=tmax, ymin=ymin, ymax=ymax, zmin=zmin, zmax=zmax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title, rti=True, XAxisAsTime=True,
|
|
|
ticksize=9, cblabel='', cbsize="1%")
|
|
|
|
|
|
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')
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
if save:
|
|
|
|
|
|
if figfile == None:
|
|
|
figfile = self.getFilename(name = self.name)
|
|
|
|
|
|
self.saveFigure(figpath, figfile)
|
|
|
|
|
|
if x[1] + (x[1]-x[0]) >= self.axesList[0].xmax:
|
|
|
self.__isConfig = False
|
|
|
|
|
|
class SpectraPlot(Figure):
|
|
|
|
|
|
__isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
WIDTHPROF = None
|
|
|
HEIGHTPROF = None
|
|
|
PREFIX = 'spc'
|
|
|
|
|
|
def __init__(self):
|
|
|
|
|
|
self.__isConfig = False
|
|
|
self.__nsubplots = 1
|
|
|
|
|
|
self.WIDTH = 250
|
|
|
self.HEIGHT = 250
|
|
|
self.WIDTHPROF = 120
|
|
|
self.HEIGHTPROF = 0
|
|
|
|
|
|
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, idfigure, nplots, wintitle, showprofile=True):
|
|
|
|
|
|
self.__showprofile = showprofile
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 1
|
|
|
colspan = 1
|
|
|
widthplot = self.WIDTH
|
|
|
heightplot = self.HEIGHT
|
|
|
if showprofile:
|
|
|
ncolspan = 3
|
|
|
colspan = 2
|
|
|
self.__nsubplots = 2
|
|
|
widthplot += self.WIDTHPROF
|
|
|
heightplot += self.HEIGHTPROF
|
|
|
|
|
|
self.createFigure(idfigure = idfigure,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = widthplot,
|
|
|
heightplot = heightplot)
|
|
|
|
|
|
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, idfigure, wintitle="", channelList=None, showprofile='True',
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None, normalize=True,
|
|
|
save=False, figpath='./', figfile=None):
|
|
|
|
|
|
"""
|
|
|
|
|
|
Input:
|
|
|
dataOut :
|
|
|
idfigure :
|
|
|
wintitle :
|
|
|
channelList :
|
|
|
showProfile :
|
|
|
xmin : None,
|
|
|
xmax : None,
|
|
|
ymin : None,
|
|
|
ymax : None,
|
|
|
zmin : None,
|
|
|
zmax : None
|
|
|
"""
|
|
|
|
|
|
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))
|
|
|
factor = 1
|
|
|
if normalize:
|
|
|
factor = dataOut.normFactor
|
|
|
x = dataOut.getVelRange(1)
|
|
|
y = dataOut.getHeiRange()
|
|
|
|
|
|
z = dataOut.data_spc[channelIndexList,:,:]/factor
|
|
|
z = numpy.where(numpy.isfinite(z), z, numpy.NAN)
|
|
|
avg = numpy.average(z, axis=1)
|
|
|
noise = dataOut.getNoise()/factor
|
|
|
|
|
|
zdB = 10*numpy.log10(z)
|
|
|
avgdB = 10*numpy.log10(avg)
|
|
|
noisedB = 10*numpy.log10(noise)
|
|
|
|
|
|
thisDatetime = dataOut.datatime
|
|
|
title = "Spectra: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
xlabel = "Velocity (m/s)"
|
|
|
ylabel = "Range (Km)"
|
|
|
|
|
|
if not self.__isConfig:
|
|
|
|
|
|
nplots = len(channelIndexList)
|
|
|
|
|
|
self.setup(idfigure=idfigure,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
showprofile=showprofile)
|
|
|
|
|
|
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.nanmin(avgdB)*0.9
|
|
|
if zmax == None: zmax = numpy.nanmax(avgdB)*0.9
|
|
|
|
|
|
self.__isConfig = True
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
for i in range(self.nplots):
|
|
|
title = "Channel %d: %4.2fdB" %(dataOut.channelList[i], noisedB[i])
|
|
|
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='')
|
|
|
|
|
|
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')
|
|
|
|
|
|
noiseline = numpy.repeat(noisedB[i], len(y))
|
|
|
axes.addpline(noiseline, y, idline=1, color="black", linestyle="dashed", lw=2)
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
if save:
|
|
|
date = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
if figfile == None:
|
|
|
figfile = self.getFilename(name = date)
|
|
|
|
|
|
self.saveFigure(figpath, figfile)
|
|
|
|
|
|
class Scope(Figure):
|
|
|
|
|
|
__isConfig = None
|
|
|
|
|
|
def __init__(self):
|
|
|
|
|
|
self.__isConfig = False
|
|
|
self.WIDTH = 600
|
|
|
self.HEIGHT = 200
|
|
|
|
|
|
def getSubplots(self):
|
|
|
|
|
|
nrow = self.nplots
|
|
|
ncol = 3
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, idfigure, nplots, wintitle):
|
|
|
|
|
|
self.nplots = nplots
|
|
|
|
|
|
self.createFigure(idfigure, wintitle)
|
|
|
|
|
|
nrow,ncol = self.getSubplots()
|
|
|
colspan = 3
|
|
|
rowspan = 1
|
|
|
|
|
|
for i in range(nplots):
|
|
|
self.addAxes(nrow, ncol, i, 0, colspan, rowspan)
|
|
|
|
|
|
|
|
|
|
|
|
def run(self, dataOut, idfigure, wintitle="", channelList=None,
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, save=False,
|
|
|
figpath='./', figfile=None):
|
|
|
|
|
|
"""
|
|
|
|
|
|
Input:
|
|
|
dataOut :
|
|
|
idfigure :
|
|
|
wintitle :
|
|
|
channelList :
|
|
|
xmin : None,
|
|
|
xmax : None,
|
|
|
ymin : None,
|
|
|
ymax : None,
|
|
|
"""
|
|
|
|
|
|
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))
|
|
|
|
|
|
x = dataOut.heightList
|
|
|
y = dataOut.data[channelIndexList,:] * numpy.conjugate(dataOut.data[channelIndexList,:])
|
|
|
y = y.real
|
|
|
|
|
|
thisDatetime = dataOut.datatime
|
|
|
title = "Scope: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
xlabel = "Range (Km)"
|
|
|
ylabel = "Intensity"
|
|
|
|
|
|
if not self.__isConfig:
|
|
|
nplots = len(channelIndexList)
|
|
|
|
|
|
self.setup(idfigure=idfigure,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle)
|
|
|
|
|
|
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)
|
|
|
|
|
|
self.__isConfig = True
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
for i in range(len(self.axesList)):
|
|
|
title = "Channel %d" %(i)
|
|
|
axes = self.axesList[i]
|
|
|
ychannel = y[i,:]
|
|
|
axes.pline(x, ychannel,
|
|
|
xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title)
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
if save:
|
|
|
date = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
if figfile == None:
|
|
|
figfile = self.getFilename(name = date)
|
|
|
|
|
|
self.saveFigure(figpath, figfile)
|
|
|
|
|
|
class ProfilePlot(Figure):
|
|
|
__isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
WIDTHPROF = None
|
|
|
HEIGHTPROF = None
|
|
|
PREFIX = 'spcprofile'
|
|
|
|
|
|
def __init__(self):
|
|
|
self.__isConfig = False
|
|
|
self.__nsubplots = 1
|
|
|
|
|
|
self.WIDTH = 300
|
|
|
self.HEIGHT = 500
|
|
|
|
|
|
def getSubplots(self):
|
|
|
ncol = 1
|
|
|
nrow = 1
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, idfigure, nplots, wintitle):
|
|
|
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 1
|
|
|
colspan = 1
|
|
|
|
|
|
self.createFigure(idfigure = idfigure,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = self.WIDTH,
|
|
|
heightplot = self.HEIGHT)
|
|
|
|
|
|
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, idfigure, wintitle="", channelList=None,
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, normalize=True,
|
|
|
save=False, figpath='./', figfile=None):
|
|
|
|
|
|
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 = 1
|
|
|
if normalize:
|
|
|
factor = dataOut.normFactor
|
|
|
y = dataOut.getHeiRange()
|
|
|
x = dataOut.data_spc[channelIndexList,:,:]/factor
|
|
|
x = numpy.where(numpy.isfinite(x), x, numpy.NAN)
|
|
|
avg = numpy.average(x, axis=1)
|
|
|
|
|
|
avgdB = 10*numpy.log10(avg)
|
|
|
|
|
|
thisDatetime = dataOut.datatime
|
|
|
title = "Power Profile"
|
|
|
xlabel = "dB"
|
|
|
ylabel = "Range (Km)"
|
|
|
|
|
|
if not self.__isConfig:
|
|
|
|
|
|
nplots = 1
|
|
|
|
|
|
self.setup(idfigure=idfigure,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle)
|
|
|
|
|
|
if ymin == None: ymin = numpy.nanmin(y)
|
|
|
if ymax == None: ymax = numpy.nanmax(y)
|
|
|
if xmin == None: xmin = numpy.nanmin(avgdB)*0.9
|
|
|
if xmax == None: xmax = numpy.nanmax(avgdB)*0.9
|
|
|
|
|
|
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(avgdB, 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()
|
|
|
|
|
|
if save:
|
|
|
date = thisDatetime.strftime("%Y%m%d")
|
|
|
if figfile == None:
|
|
|
figfile = self.getFilename(name = date)
|
|
|
|
|
|
self.saveFigure(figpath, figfile)
|
|
|
|
|
|
class CoherenceMap(Figure):
|
|
|
__isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
WIDTHPROF = None
|
|
|
HEIGHTPROF = None
|
|
|
PREFIX = 'cmap'
|
|
|
__missing = 1E30
|
|
|
|
|
|
def __init__(self):
|
|
|
self.timerange = 2*60*60
|
|
|
self.__isConfig = False
|
|
|
self.__nsubplots = 1
|
|
|
|
|
|
self.WIDTH = 800
|
|
|
self.HEIGHT = 200
|
|
|
self.WIDTHPROF = 120
|
|
|
self.HEIGHTPROF = 0
|
|
|
self.x_buffer = None
|
|
|
self.coherence_buffer = None
|
|
|
self.phase_buffer = None
|
|
|
|
|
|
def getSubplots(self):
|
|
|
ncol = 1
|
|
|
nrow = self.nplots*2
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, idfigure, nplots, wintitle, showprofile=True):
|
|
|
self.__showprofile = showprofile
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 1
|
|
|
colspan = 1
|
|
|
if showprofile:
|
|
|
ncolspan = 7
|
|
|
colspan = 6
|
|
|
self.__nsubplots = 2
|
|
|
|
|
|
self.createFigure(idfigure = idfigure,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = self.WIDTH + self.WIDTHPROF,
|
|
|
heightplot = self.HEIGHT + self.HEIGHTPROF)
|
|
|
|
|
|
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, idfigure, wintitle="", pairsList=None, showprofile='True',
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None,
|
|
|
timerange=None,
|
|
|
save=False, figpath='./', figfile=None,
|
|
|
coherence_cmap='jet', phase_cmap='RdBu_r'):
|
|
|
|
|
|
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 timerange != None:
|
|
|
self.timerange = timerange
|
|
|
|
|
|
if pairsIndexList == []:
|
|
|
return
|
|
|
|
|
|
if len(pairsIndexList) > 4:
|
|
|
pairsIndexList = pairsIndexList[0:4]
|
|
|
|
|
|
tmin = None
|
|
|
tmax = None
|
|
|
x = dataOut.getTimeRange()
|
|
|
y = dataOut.getHeiRange()
|
|
|
|
|
|
thisDatetime = dataOut.datatime
|
|
|
title = "CoherenceMap: %s" %(thisDatetime.strftime("%d-%b-%Y"))
|
|
|
xlabel = ""
|
|
|
ylabel = "Range (Km)"
|
|
|
|
|
|
if not self.__isConfig:
|
|
|
nplots = len(pairsIndexList)
|
|
|
self.setup(idfigure=idfigure,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
showprofile=showprofile)
|
|
|
|
|
|
tmin, tmax = self.getTimeLim(x, xmin, xmax)
|
|
|
if ymin == None: ymin = numpy.nanmin(y)
|
|
|
if ymax == None: ymax = numpy.nanmax(y)
|
|
|
|
|
|
self.name = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
self.x_buffer = numpy.array([])
|
|
|
self.coherence_buffer = numpy.array([])
|
|
|
self.phase_buffer = numpy.array([])
|
|
|
self.__isConfig = True
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
|
|
|
pairArray = numpy.array(dataOut.pairsList)
|
|
|
pairArray = pairArray[pairsIndexList]
|
|
|
pair0ids = pairArray[:,0]
|
|
|
pair1ids = pairArray[:,1]
|
|
|
|
|
|
coherenceComplex = dataOut.data_cspc[pairsIndexList,:,:]/numpy.sqrt(dataOut.data_spc[pair0ids,:,:]*dataOut.data_spc[pair1ids,:,:])
|
|
|
avgcoherenceComplex = numpy.average(coherenceComplex, axis=1)
|
|
|
coherence = numpy.abs(avgcoherenceComplex)
|
|
|
|
|
|
phase = numpy.arctan2(avgcoherenceComplex.imag, avgcoherenceComplex.real)*180/numpy.pi
|
|
|
|
|
|
if len(self.coherence_buffer)==0:
|
|
|
self.coherence_buffer = coherence
|
|
|
self.phase_buffer = phase
|
|
|
newxdim = 1
|
|
|
newydim = -1
|
|
|
else:
|
|
|
if x[0]>self.x_buffer[-1]:
|
|
|
gap = coherence.copy()
|
|
|
gap[:] = self.__missing
|
|
|
self.coherence_buffer = numpy.hstack((self.coherence_buffer, gap))
|
|
|
self.phase_buffer = numpy.hstack((self.phase_buffer, gap))
|
|
|
|
|
|
self.coherence_buffer = numpy.hstack((self.coherence_buffer, coherence))
|
|
|
self.phase_buffer = numpy.hstack((self.phase_buffer, phase))
|
|
|
newxdim = -1
|
|
|
newydim = len(y)
|
|
|
|
|
|
self.x_buffer = numpy.hstack((self.x_buffer, x))
|
|
|
|
|
|
self.coherence_buffer = numpy.ma.masked_inside(self.coherence_buffer,0.99*self.__missing,1.01*self.__missing)
|
|
|
self.phase_buffer = numpy.ma.masked_inside(self.phase_buffer,0.99*self.__missing,1.01*self.__missing)
|
|
|
|
|
|
|
|
|
for i in range(self.nplots):
|
|
|
counter = 0
|
|
|
z = self.coherence_buffer[i,:].reshape((newxdim,newydim))
|
|
|
title = "Coherence %d%d: %s" %(pair0ids[i], pair1ids[i], thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
axes = self.axesList[i*self.__nsubplots*2]
|
|
|
axes.pcolor(self.x_buffer, y, z,
|
|
|
xmin=tmin, xmax=tmax, ymin=ymin, ymax=ymax, zmin=0, zmax=1,
|
|
|
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[i,:], y,
|
|
|
xmin=0, xmax=1, ymin=ymin, ymax=ymax,
|
|
|
xlabel='', ylabel='', title='', ticksize=7,
|
|
|
ytick_visible=False, nxticks=5,
|
|
|
grid='x')
|
|
|
|
|
|
counter += 1
|
|
|
|
|
|
z = self.phase_buffer[i,:].reshape((newxdim,newydim))
|
|
|
|
|
|
title = "Phase %d%d: %s" %(pair0ids[i], pair1ids[i], thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
axes = self.axesList[i*self.__nsubplots*2 + counter]
|
|
|
axes.pcolor(self.x_buffer, y, z,
|
|
|
xmin=tmin, xmax=tmax, ymin=ymin, ymax=ymax, zmin=-180, zmax=180,
|
|
|
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[i,:], y,
|
|
|
xmin=-180, xmax=180, ymin=ymin, ymax=ymax,
|
|
|
xlabel='', ylabel='', title='', ticksize=7,
|
|
|
ytick_visible=False, nxticks=4,
|
|
|
grid='x')
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
if save:
|
|
|
|
|
|
if figfile == None:
|
|
|
figfile = self.getFilename(name = self.name)
|
|
|
|
|
|
self.saveFigure(figpath, figfile)
|
|
|
|
|
|
if x[1] + (x[1]-x[0]) >= self.axesList[0].xmax:
|
|
|
self.__isConfig = False
|
|
|
|
|
|
class RTIfromNoise(Figure):
|
|
|
|
|
|
__isConfig = None
|
|
|
__nsubplots = None
|
|
|
|
|
|
PREFIX = 'rtinoise'
|
|
|
|
|
|
def __init__(self):
|
|
|
|
|
|
self.timerange = 24*60*60
|
|
|
self.__isConfig = False
|
|
|
self.__nsubplots = 1
|
|
|
|
|
|
self.WIDTH = 820
|
|
|
self.HEIGHT = 200
|
|
|
self.WIDTHPROF = 120
|
|
|
self.HEIGHTPROF = 0
|
|
|
self.xdata = None
|
|
|
self.ydata = None
|
|
|
|
|
|
def getSubplots(self):
|
|
|
|
|
|
ncol = 1
|
|
|
nrow = 1
|
|
|
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, idfigure, nplots, wintitle, showprofile=True):
|
|
|
|
|
|
self.__showprofile = showprofile
|
|
|
self.nplots = nplots
|
|
|
|
|
|
ncolspan = 7
|
|
|
colspan = 6
|
|
|
self.__nsubplots = 2
|
|
|
|
|
|
self.createFigure(idfigure = idfigure,
|
|
|
wintitle = wintitle,
|
|
|
widthplot = self.WIDTH+self.WIDTHPROF,
|
|
|
heightplot = self.HEIGHT+self.HEIGHTPROF)
|
|
|
|
|
|
nrow, ncol = self.getSubplots()
|
|
|
|
|
|
self.addAxes(nrow, ncol*ncolspan, 0, 0, colspan, 1)
|
|
|
|
|
|
|
|
|
def run(self, dataOut, idfigure, wintitle="", channelList=None, showprofile='True',
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, normalize=True,
|
|
|
timerange=None,
|
|
|
save=False, figpath='./', figfile=None):
|
|
|
|
|
|
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))
|
|
|
|
|
|
if timerange != None:
|
|
|
self.timerange = timerange
|
|
|
|
|
|
tmin = None
|
|
|
tmax = None
|
|
|
x = dataOut.getTimeRange()
|
|
|
y = dataOut.getHeiRange()
|
|
|
factor = 1
|
|
|
if normalize:
|
|
|
factor = dataOut.normFactor
|
|
|
noise = dataOut.getNoise()/factor
|
|
|
noisedB = 10*numpy.log10(noise)
|
|
|
|
|
|
thisDatetime = dataOut.datatime
|
|
|
title = "RTI Noise: %s" %(thisDatetime.strftime("%d-%b-%Y"))
|
|
|
xlabel = ""
|
|
|
ylabel = "Range (Km)"
|
|
|
|
|
|
if not self.__isConfig:
|
|
|
|
|
|
nplots = 1
|
|
|
|
|
|
self.setup(idfigure=idfigure,
|
|
|
nplots=nplots,
|
|
|
wintitle=wintitle,
|
|
|
showprofile=showprofile)
|
|
|
|
|
|
tmin, tmax = self.getTimeLim(x, xmin, xmax)
|
|
|
if ymin == None: ymin = numpy.nanmin(noisedB)
|
|
|
if ymax == None: ymax = numpy.nanmax(noisedB)
|
|
|
|
|
|
self.name = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
self.__isConfig = True
|
|
|
|
|
|
self.xdata = numpy.array([])
|
|
|
self.ydata = numpy.array([])
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
|
|
|
title = "RTI Noise %s" %(thisDatetime.strftime("%d-%b-%Y"))
|
|
|
|
|
|
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[channelIndexList].reshape(-1,1)
|
|
|
else:
|
|
|
self.ydata = numpy.hstack((self.ydata, noisedB[channelIndexList].reshape(-1,1)))
|
|
|
|
|
|
|
|
|
axes.pmultilineyaxis(x=self.xdata, y=self.ydata,
|
|
|
xmin=tmin, xmax=tmax, ymin=ymin, ymax=ymax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title, legendlabels=legendlabels, marker='x', markersize=8, linestyle="solid",
|
|
|
XAxisAsTime=True
|
|
|
)
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
if save:
|
|
|
|
|
|
if figfile == None:
|
|
|
figfile = self.getFilename(name = self.name)
|
|
|
|
|
|
self.saveFigure(figpath, figfile)
|
|
|
|
|
|
if x[1] + (x[1]-x[0]) >= self.axesList[0].xmax:
|
|
|
self.__isConfig = False
|
|
|
del self.xdata
|
|
|
del self.ydata
|
|
|
|
