|
|
'''
|
|
|
Created on Jul 9, 2014
|
|
|
|
|
|
@author: roj-idl71
|
|
|
'''
|
|
|
import os
|
|
|
import datetime
|
|
|
import numpy
|
|
|
|
|
|
from schainpy.model.graphics.jroplot_base import Plot, plt
|
|
|
|
|
|
|
|
|
class ScopePlot(Plot):
|
|
|
|
|
|
'''
|
|
|
Plot for Scope
|
|
|
'''
|
|
|
|
|
|
CODE = 'scope'
|
|
|
plot_type = 'scatter'
|
|
|
|
|
|
def setup(self):
|
|
|
|
|
|
self.xaxis = 'Range (Km)'
|
|
|
self.nplots = len(self.data.channels)
|
|
|
self.nrows = int(numpy.ceil(self.nplots/2))
|
|
|
self.ncols = int(numpy.ceil(self.nplots/self.nrows))
|
|
|
self.ylabel = 'Intensity [dB]'
|
|
|
self.titles = ['Channel '+str(self.data.channels[i])+" " for i in self.data.channels]
|
|
|
self.colorbar = False
|
|
|
self.width = 6
|
|
|
self.height = 4
|
|
|
|
|
|
def update(self, dataOut):
|
|
|
|
|
|
data = {}
|
|
|
meta = {
|
|
|
'nProfiles': dataOut.nProfiles,
|
|
|
'flagDataAsBlock': dataOut.flagDataAsBlock,
|
|
|
'profileIndex': dataOut.profileIndex,
|
|
|
}
|
|
|
if self.CODE == 'scope':
|
|
|
data[self.CODE] = dataOut.data
|
|
|
elif self.CODE == 'pp_power':
|
|
|
data[self.CODE] = dataOut.dataPP_POWER
|
|
|
elif self.CODE == 'pp_signal':
|
|
|
data[self.CODE] = dataOut.dataPP_POW
|
|
|
elif self.CODE == 'pp_velocity':
|
|
|
data[self.CODE] = dataOut.dataPP_DOP
|
|
|
elif self.CODE == 'pp_specwidth':
|
|
|
data[self.CODE] = dataOut.dataPP_WIDTH
|
|
|
|
|
|
return data, meta
|
|
|
|
|
|
def plot_iq(self, x, y, channelIndexList, thisDatetime, wintitle):
|
|
|
|
|
|
yreal = y[channelIndexList,:].real
|
|
|
yimag = y[channelIndexList,:].imag
|
|
|
Maintitle = wintitle + " Scope: %s" %(thisDatetime.strftime("%d-%b-%Y"))
|
|
|
self.xlabel = "Range (Km)"
|
|
|
self.ylabel = "Intensity - IQ"
|
|
|
|
|
|
self.y = yreal
|
|
|
self.x = x
|
|
|
|
|
|
for i,ax in enumerate(self.axes):
|
|
|
title = "Channel %d" %(i)
|
|
|
if ax.firsttime:
|
|
|
self.xmin = min(x)
|
|
|
self.xmax = max(x)
|
|
|
ax.plt_r = ax.plot(x, yreal[i,:], color='b')[0]
|
|
|
ax.plt_i = ax.plot(x, yimag[i,:], color='r')[0]
|
|
|
else:
|
|
|
ax.plt_r.set_data(x, yreal[i,:])
|
|
|
ax.plt_i.set_data(x, yimag[i,:])
|
|
|
plt.suptitle(Maintitle)
|
|
|
|
|
|
def plot_power(self, x, y, channelIndexList, thisDatetime, wintitle):
|
|
|
y = y[channelIndexList,:] * numpy.conjugate(y[channelIndexList,:])
|
|
|
yreal = y.real
|
|
|
yreal = 10*numpy.log10(yreal)
|
|
|
self.y = yreal
|
|
|
title = wintitle + " Power: %s" %(thisDatetime.strftime("%d-%b-%Y"))
|
|
|
self.xlabel = "Range (Km)"
|
|
|
self.ylabel = "Intensity [dB]"
|
|
|
|
|
|
|
|
|
self.titles[0] = title
|
|
|
|
|
|
for i,ax in enumerate(self.axes):
|
|
|
title = "Channel %d" %(i)
|
|
|
ychannel = yreal[i,:]
|
|
|
|
|
|
if ax.firsttime:
|
|
|
self.xmin = min(x)
|
|
|
self.xmax = max(x)
|
|
|
ax.plt_r = ax.plot(x, ychannel)[0]
|
|
|
else:
|
|
|
ax.plt_r.set_data(x, ychannel)
|
|
|
|
|
|
|
|
|
def plot_weatherpower(self, x, y, channelIndexList, thisDatetime, wintitle):
|
|
|
|
|
|
|
|
|
y = y[channelIndexList,:]
|
|
|
yreal = y.real
|
|
|
yreal = 10*numpy.log10(yreal)
|
|
|
self.y = yreal
|
|
|
title = wintitle + " Scope: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
self.xlabel = "Range (Km)"
|
|
|
self.ylabel = "Intensity"
|
|
|
self.xmin = min(x)
|
|
|
self.xmax = max(x)
|
|
|
|
|
|
self.titles[0] =title
|
|
|
for i,ax in enumerate(self.axes):
|
|
|
title = "Channel %d" %(i)
|
|
|
|
|
|
ychannel = yreal[i,:]
|
|
|
|
|
|
if ax.firsttime:
|
|
|
ax.plt_r = ax.plot(x, ychannel)[0]
|
|
|
else:
|
|
|
#pass
|
|
|
ax.plt_r.set_data(x, ychannel)
|
|
|
|
|
|
def plot_weathervelocity(self, x, y, channelIndexList, thisDatetime, wintitle):
|
|
|
|
|
|
x = x[channelIndexList,:]
|
|
|
yreal = y
|
|
|
self.y = yreal
|
|
|
title = wintitle + " Scope: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
self.xlabel = "Velocity (m/s)"
|
|
|
self.ylabel = "Range (Km)"
|
|
|
self.xmin = numpy.min(x)
|
|
|
self.xmax = numpy.max(x)
|
|
|
self.titles[0] =title
|
|
|
for i,ax in enumerate(self.axes):
|
|
|
title = "Channel %d" %(i)
|
|
|
xchannel = x[i,:]
|
|
|
if ax.firsttime:
|
|
|
ax.plt_r = ax.plot(xchannel, yreal)[0]
|
|
|
else:
|
|
|
#pass
|
|
|
ax.plt_r.set_data(xchannel, yreal)
|
|
|
|
|
|
def plot_weatherspecwidth(self, x, y, channelIndexList, thisDatetime, wintitle):
|
|
|
|
|
|
x = x[channelIndexList,:]
|
|
|
yreal = y
|
|
|
self.y = yreal
|
|
|
title = wintitle + " Scope: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
self.xlabel = "width "
|
|
|
self.ylabel = "Range (Km)"
|
|
|
self.xmin = numpy.min(x)
|
|
|
self.xmax = numpy.max(x)
|
|
|
self.titles[0] =title
|
|
|
for i,ax in enumerate(self.axes):
|
|
|
title = "Channel %d" %(i)
|
|
|
xchannel = x[i,:]
|
|
|
if ax.firsttime:
|
|
|
ax.plt_r = ax.plot(xchannel, yreal)[0]
|
|
|
else:
|
|
|
#pass
|
|
|
ax.plt_r.set_data(xchannel, yreal)
|
|
|
|
|
|
def plot(self):
|
|
|
if self.channels:
|
|
|
channels = self.channels
|
|
|
else:
|
|
|
channels = self.data.channels
|
|
|
|
|
|
thisDatetime = datetime.datetime.utcfromtimestamp(self.data.times[-1])
|
|
|
|
|
|
scope = self.data[-1][self.CODE]
|
|
|
|
|
|
if self.data.flagDataAsBlock:
|
|
|
|
|
|
for i in range(self.data.nProfiles):
|
|
|
|
|
|
wintitle1 = " [Profile = %d] " %i
|
|
|
if self.CODE =="scope":
|
|
|
if self.type == "power":
|
|
|
self.plot_power(self.data.yrange,
|
|
|
scope[:,i,:],
|
|
|
channels,
|
|
|
thisDatetime,
|
|
|
wintitle1
|
|
|
)
|
|
|
|
|
|
if self.type == "iq":
|
|
|
self.plot_iq(self.data.yrange,
|
|
|
scope[:,i,:],
|
|
|
channels,
|
|
|
thisDatetime,
|
|
|
wintitle1
|
|
|
)
|
|
|
if self.CODE=="pp_power":
|
|
|
self.plot_weatherpower(self.data.yrange,
|
|
|
scope[:,i,:],
|
|
|
channels,
|
|
|
thisDatetime,
|
|
|
wintitle
|
|
|
)
|
|
|
if self.CODE=="pp_signal":
|
|
|
self.plot_weatherpower(self.data.yrange,
|
|
|
scope[:,i,:],
|
|
|
channels,
|
|
|
thisDatetime,
|
|
|
wintitle
|
|
|
)
|
|
|
if self.CODE=="pp_velocity":
|
|
|
self.plot_weathervelocity(scope[:,i,:],
|
|
|
self.data.yrange,
|
|
|
channels,
|
|
|
thisDatetime,
|
|
|
wintitle
|
|
|
)
|
|
|
if self.CODE=="pp_spcwidth":
|
|
|
self.plot_weatherspecwidth(scope[:,i,:],
|
|
|
self.data.yrange,
|
|
|
channels,
|
|
|
thisDatetime,
|
|
|
wintitle
|
|
|
)
|
|
|
else:
|
|
|
wintitle = " [Profile = %d] " %self.data.profileIndex
|
|
|
if self.CODE== "scope":
|
|
|
if self.type == "power":
|
|
|
self.plot_power(self.data.yrange,
|
|
|
scope,
|
|
|
channels,
|
|
|
thisDatetime,
|
|
|
wintitle
|
|
|
)
|
|
|
|
|
|
if self.type == "iq":
|
|
|
self.plot_iq(self.data.yrange,
|
|
|
scope,
|
|
|
channels,
|
|
|
thisDatetime,
|
|
|
wintitle
|
|
|
)
|
|
|
if self.CODE=="pp_power":
|
|
|
self.plot_weatherpower(self.data.yrange,
|
|
|
scope,
|
|
|
channels,
|
|
|
thisDatetime,
|
|
|
wintitle
|
|
|
)
|
|
|
if self.CODE=="pp_signal":
|
|
|
self.plot_weatherpower(self.data.yrange,
|
|
|
scope,
|
|
|
channels,
|
|
|
thisDatetime,
|
|
|
wintitle
|
|
|
)
|
|
|
if self.CODE=="pp_velocity":
|
|
|
self.plot_weathervelocity(scope,
|
|
|
self.data.yrange,
|
|
|
channels,
|
|
|
thisDatetime,
|
|
|
wintitle
|
|
|
)
|
|
|
if self.CODE=="pp_specwidth":
|
|
|
self.plot_weatherspecwidth(scope,
|
|
|
self.data.yrange,
|
|
|
channels,
|
|
|
thisDatetime,
|
|
|
wintitle
|
|
|
)
|
|
|
|
|
|
|
|
|
class PulsepairPowerPlot(ScopePlot):
|
|
|
'''
|
|
|
Plot for P= S+N
|
|
|
'''
|
|
|
|
|
|
CODE = 'pp_power'
|
|
|
plot_type = 'scatter'
|
|
|
|
|
|
class PulsepairVelocityPlot(ScopePlot):
|
|
|
'''
|
|
|
Plot for VELOCITY
|
|
|
'''
|
|
|
CODE = 'pp_velocity'
|
|
|
plot_type = 'scatter'
|
|
|
|
|
|
class PulsepairSpecwidthPlot(ScopePlot):
|
|
|
'''
|
|
|
Plot for WIDTH
|
|
|
'''
|
|
|
CODE = 'pp_specwidth'
|
|
|
plot_type = 'scatter'
|
|
|
|
|
|
class PulsepairSignalPlot(ScopePlot):
|
|
|
'''
|
|
|
Plot for S
|
|
|
'''
|
|
|
|
|
|
CODE = 'pp_signal'
|
|
|
plot_type = 'scatter'
|
|
|
|
