|
|
'''
|
|
|
Created on Jul 9, 2014
|
|
|
|
|
|
@author: roj-idl71
|
|
|
'''
|
|
|
import os
|
|
|
import datetime
|
|
|
import numpy
|
|
|
|
|
|
from figure import Figure
|
|
|
from plotting_codes import *
|
|
|
|
|
|
class Scope(Figure):
|
|
|
|
|
|
isConfig = None
|
|
|
|
|
|
def __init__(self, **kwargs):
|
|
|
Figure.__init__(self, **kwargs)
|
|
|
self.isConfig = False
|
|
|
self.WIDTH = 300
|
|
|
self.HEIGHT = 200
|
|
|
self.counter_imagwr = 0
|
|
|
|
|
|
def getSubplots(self):
|
|
|
|
|
|
nrow = self.nplots
|
|
|
ncol = 3
|
|
|
return nrow, ncol
|
|
|
|
|
|
def setup(self, id, nplots, wintitle, show):
|
|
|
|
|
|
self.nplots = nplots
|
|
|
|
|
|
self.createFigure(id=id,
|
|
|
wintitle=wintitle,
|
|
|
show=show)
|
|
|
|
|
|
nrow,ncol = self.getSubplots()
|
|
|
colspan = 3
|
|
|
rowspan = 1
|
|
|
|
|
|
for i in range(nplots):
|
|
|
self.addAxes(nrow, ncol, i, 0, colspan, rowspan)
|
|
|
|
|
|
def plot_iq(self, x, y, id, channelIndexList, thisDatetime, wintitle, show, xmin, xmax, ymin, ymax):
|
|
|
yreal = y[channelIndexList,:].real
|
|
|
yimag = y[channelIndexList,:].imag
|
|
|
|
|
|
title = wintitle + " Scope: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
xlabel = "Range (Km)"
|
|
|
ylabel = "Intensity - IQ"
|
|
|
|
|
|
if not self.isConfig:
|
|
|
nplots = len(channelIndexList)
|
|
|
|
|
|
self.setup(id=id,
|
|
|
nplots=nplots,
|
|
|
wintitle='',
|
|
|
show=show)
|
|
|
|
|
|
if xmin == None: xmin = numpy.nanmin(x)
|
|
|
if xmax == None: xmax = numpy.nanmax(x)
|
|
|
if ymin == None: ymin = min(numpy.nanmin(yreal),numpy.nanmin(yimag))
|
|
|
if ymax == None: ymax = max(numpy.nanmax(yreal),numpy.nanmax(yimag))
|
|
|
|
|
|
self.isConfig = True
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
for i in range(len(self.axesList)):
|
|
|
title = "Channel %d" %(i)
|
|
|
axes = self.axesList[i]
|
|
|
|
|
|
axes.pline(x, yreal[i,:],
|
|
|
xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title)
|
|
|
|
|
|
axes.addpline(x, yimag[i,:], idline=1, color="red", linestyle="solid", lw=2)
|
|
|
|
|
|
def plot_power(self, x, y, id, channelIndexList, thisDatetime, wintitle, show, xmin, xmax, ymin, ymax):
|
|
|
y = y[channelIndexList,:] * numpy.conjugate(y[channelIndexList,:])
|
|
|
yreal = y.real
|
|
|
|
|
|
title = wintitle + " Scope: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
xlabel = "Range (Km)"
|
|
|
ylabel = "Intensity"
|
|
|
|
|
|
if not self.isConfig:
|
|
|
nplots = len(channelIndexList)
|
|
|
|
|
|
self.setup(id=id,
|
|
|
nplots=nplots,
|
|
|
wintitle='',
|
|
|
show=show)
|
|
|
|
|
|
if xmin == None: xmin = numpy.nanmin(x)
|
|
|
if xmax == None: xmax = numpy.nanmax(x)
|
|
|
if ymin == None: ymin = numpy.nanmin(yreal)
|
|
|
if ymax == None: ymax = numpy.nanmax(yreal)
|
|
|
|
|
|
self.isConfig = True
|
|
|
|
|
|
self.setWinTitle(title)
|
|
|
|
|
|
for i in range(len(self.axesList)):
|
|
|
title = "Channel %d" %(i)
|
|
|
axes = self.axesList[i]
|
|
|
ychannel = yreal[i,:]
|
|
|
axes.pline(x, ychannel,
|
|
|
xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax,
|
|
|
xlabel=xlabel, ylabel=ylabel, title=title)
|
|
|
|
|
|
|
|
|
def run(self, dataOut, id, wintitle="", channelList=None,
|
|
|
xmin=None, xmax=None, ymin=None, ymax=None, save=False,
|
|
|
figpath='./', figfile=None, show=True, wr_period=1,
|
|
|
ftp=False, server=None, folder=None, username=None, password=None, type='power', **kwargs):
|
|
|
|
|
|
"""
|
|
|
|
|
|
Input:
|
|
|
dataOut :
|
|
|
id :
|
|
|
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))
|
|
|
|
|
|
thisDatetime = datetime.datetime.utcfromtimestamp(dataOut.getTimeRange()[0])
|
|
|
|
|
|
if dataOut.flagDataAsBlock:
|
|
|
|
|
|
for i in range(dataOut.nProfiles):
|
|
|
|
|
|
wintitle1 = wintitle + " [Profile = %d] " %i
|
|
|
|
|
|
if type == "power":
|
|
|
self.plot_power(dataOut.heightList,
|
|
|
dataOut.data[:,i,:],
|
|
|
id,
|
|
|
channelIndexList,
|
|
|
thisDatetime,
|
|
|
wintitle1,
|
|
|
show,
|
|
|
xmin,
|
|
|
xmax,
|
|
|
ymin,
|
|
|
ymax)
|
|
|
|
|
|
if type == "iq":
|
|
|
self.plot_iq(dataOut.heightList,
|
|
|
dataOut.data[:,i,:],
|
|
|
id,
|
|
|
channelIndexList,
|
|
|
thisDatetime,
|
|
|
wintitle1,
|
|
|
show,
|
|
|
xmin,
|
|
|
xmax,
|
|
|
ymin,
|
|
|
ymax)
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
str_datetime = thisDatetime.strftime("%Y%m%d_%H%M%S")
|
|
|
figfile = self.getFilename(name = str_datetime) + "_" + str(i)
|
|
|
|
|
|
self.save(figpath=figpath,
|
|
|
figfile=figfile,
|
|
|
save=save,
|
|
|
ftp=ftp,
|
|
|
wr_period=wr_period,
|
|
|
thisDatetime=thisDatetime)
|
|
|
|
|
|
else:
|
|
|
wintitle += " [Profile = %d] " %dataOut.profileIndex
|
|
|
|
|
|
if type == "power":
|
|
|
self.plot_power(dataOut.heightList,
|
|
|
dataOut.data,
|
|
|
id,
|
|
|
channelIndexList,
|
|
|
thisDatetime,
|
|
|
wintitle,
|
|
|
show,
|
|
|
xmin,
|
|
|
xmax,
|
|
|
ymin,
|
|
|
ymax)
|
|
|
|
|
|
if type == "iq":
|
|
|
self.plot_iq(dataOut.heightList,
|
|
|
dataOut.data,
|
|
|
id,
|
|
|
channelIndexList,
|
|
|
thisDatetime,
|
|
|
wintitle,
|
|
|
show,
|
|
|
xmin,
|
|
|
xmax,
|
|
|
ymin,
|
|
|
ymax)
|
|
|
|
|
|
self.draw()
|
|
|
|
|
|
str_datetime = thisDatetime.strftime("%Y%m%d_%H%M%S") + "_" + str(dataOut.profileIndex)
|
|
|
figfile = self.getFilename(name = str_datetime)
|
|
|
|
|
|
self.save(figpath=figpath,
|
|
|
figfile=figfile,
|
|
|
save=save,
|
|
|
ftp=ftp,
|
|
|
wr_period=wr_period,
|
|
|
thisDatetime=thisDatetime)
|
|
|
|
|
|
|
|
|
class VoltACFPLot(Figure):
|
|
|
|
|
|
isConfig = None
|
|
|
__nsubplots = None
|
|
|
PREFIX = 'voltacf'
|
|
|
|
|
|
def __init__(self, **kwargs):
|
|
|
Figure.__init__(self,**kwargs)
|
|
|
self.isConfig = False
|
|
|
self.__nsubplots = 1
|
|
|
self.PLOT_CODE = VOLT_ACF_CODE
|
|
|
self.WIDTH = 900
|
|
|
self.HEIGHT = 700
|
|
|
self.counter_imagwr= 0
|
|
|
self.FTP_WEI = None
|
|
|
self.EXP_CODE = None
|
|
|
self.SUB_EXP_CODE = None
|
|
|
self.PLOT_POS = None
|
|
|
|
|
|
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 , channel =None, nSamples = None,
|
|
|
nSampleList= None, resolutionFactor=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, ftp_wei=0 , exp_code=0,sub_exp_code=0,plot_pos=0,
|
|
|
xaxis="time"):
|
|
|
|
|
|
channel0 = channel
|
|
|
nSamples = nSamples
|
|
|
resFactor = resolutionFactor
|
|
|
|
|
|
if nSamples == None:
|
|
|
nSamples = 20
|
|
|
|
|
|
if resFactor == None:
|
|
|
resFactor = 5
|
|
|
|
|
|
if channel0 == None:
|
|
|
channel0 = 0
|
|
|
else:
|
|
|
if channel0 not in dataOut.channelList:
|
|
|
raise ValueError, "Channel %d is not in %s dataOut.channelList"%(channel0, dataOut.channelList)
|
|
|
|
|
|
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()
|
|
|
#print y, dataOut.heightList[0]
|
|
|
#print "pause"
|
|
|
#import time
|
|
|
#time.sleep(10)
|
|
|
deltaHeight = dataOut.heightList[1]-dataOut.heightList[0]
|
|
|
z = dataOut.data
|
|
|
|
|
|
shape = dataOut.data.shape
|
|
|
hei_index = numpy.arange(shape[2])
|
|
|
hei_plot = numpy.arange(nSamples)*resFactor
|
|
|
|
|
|
if nSampleList is not None:
|
|
|
for nsample in nSampleList:
|
|
|
if nsample not in dataOut.heightList/deltaHeight:
|
|
|
print "Lista available : %s "%(dataOut.heightList/deltaHeight)
|
|
|
raise ValueError, "nsample %d is not in %s dataOut.heightList"%(nsample,dataOut.heightList)
|
|
|
|
|
|
if nSampleList is not None:
|
|
|
hei_plot = numpy.array(nSampleList)*resFactor
|
|
|
|
|
|
if hei_plot[-1] >= hei_index[-1]:
|
|
|
print ("La cantidad de puntos en altura es %d y la resolucion es %f Km"%(hei_plot.shape[0],deltaHeight*resFactor ))
|
|
|
raise ValueError, "resFactor %d multiplicado por el valor de %d nSamples es mayor a %d cantidad total de puntos"%(resFactor,nSamples,hei_index[-1])
|
|
|
|
|
|
#escalamiento -1 a 1 a resolucion (factor de resolucion en altura)* deltaHeight
|
|
|
#min = numpy.min(z[0,:,0])
|
|
|
#max =numpy.max(z[0,:,0])
|
|
|
for i in range(shape[0]):
|
|
|
for j in range(shape[2]):
|
|
|
min = numpy.min(z[i,:,j])
|
|
|
max = numpy.max(z[i,:,j])
|
|
|
z[i,:,j]= (((z[i,:,j]-min)/(max-min))*deltaHeight*resFactor + j*deltaHeight+dataOut.heightList[0])
|
|
|
|
|
|
|
|
|
if xaxis == "time":
|
|
|
x = dataOut.getAcfRange()*1000
|
|
|
zdB = z[channel0,:,hei_plot]
|
|
|
xlabel = "Time (ms)"
|
|
|
ylabel = "VOLT_ACF"
|
|
|
|
|
|
|
|
|
thisDatetime = datetime.datetime.utcfromtimestamp(dataOut.getTimeRange()[0])
|
|
|
title = wintitle + "VOLT ACF Plot Ch %s %s" %(channel0,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)
|
|
|
|
|
|
print ("El parametro resFactor es %d y la resolucion en altura es %f"%(resFactor,deltaHeight ))
|
|
|
print ("La cantidad de puntos en altura es %d y la nueva resolucion es %f Km"%(hei_plot.shape[0],deltaHeight*resFactor ))
|
|
|
print ("La altura maxima es %d Km"%(hei_plot[-1]*deltaHeight ))
|
|
|
|
|
|
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)
|
|
|
|
|
|
title = "VOLT ACF Plot: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
axes = self.axesList[0]
|
|
|
|
|
|
legendlabels = ["Range = %dKm" %y[i] for i in hei_plot]
|
|
|
|
|
|
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()
|
|
|
|
|
|
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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
