|
|
import numpy
|
|
|
import datetime
|
|
|
import time
|
|
|
from schainPlot import *
|
|
|
|
|
|
class SpcFigure(Figure):
|
|
|
overplot = 0
|
|
|
xw = 800
|
|
|
yw = 650
|
|
|
showprofile = False
|
|
|
|
|
|
def __init__(self, idfigure, nframes, wintitle, driver, colormap, colorbar, showprofile):
|
|
|
Figure.__init__(self,idfigure, nframes, wintitle, self.xw, self.yw, self.overplot, driver, colormap, colorbar)
|
|
|
|
|
|
self.showprofile = showprofile
|
|
|
|
|
|
def getSubplots(self):
|
|
|
ncolumns = int(numpy.sqrt(self.nframes)+0.9)
|
|
|
nrows = int(self.nframes*1./ncolumns + 0.9)
|
|
|
|
|
|
return nrows, ncolumns
|
|
|
|
|
|
def setParms(self, data, x, y, xmin, xmax, ymin, ymax, minvalue, maxvalue, *args):
|
|
|
|
|
|
if xmin == None: xmin = numpy.min(x)
|
|
|
if xmax == None: xmax = numpy.max(x)
|
|
|
if ymin == None: ymin = numpy.min(y)
|
|
|
if ymax == None: ymax = numpy.max(y)
|
|
|
if minvalue == None: minvalue = 20.
|
|
|
if maxvalue == None: maxvalue = 90.
|
|
|
|
|
|
self.xmin = xmin
|
|
|
self.xmax = xmax
|
|
|
self.minrange = ymin
|
|
|
self.maxrange = ymax
|
|
|
self.ymin = ymin
|
|
|
self.ymax = ymax
|
|
|
self.minvalue = minvalue
|
|
|
self.maxvalue = maxvalue
|
|
|
|
|
|
|
|
|
def changeXRange(self, *args):
|
|
|
pass
|
|
|
|
|
|
def createFrames(self):
|
|
|
for frame in range(self.nframes):
|
|
|
frameObj = SpcFrame(self.drvObj,frame + 1, self.colorbar, self.showprofile)
|
|
|
self.frameObjList.append(frameObj)
|
|
|
|
|
|
class SpcFrame(Frame):
|
|
|
def __init__(self,drvObj,idframe,colorbar,showprofile):
|
|
|
self.drvObj = drvObj
|
|
|
self.idframe = idframe
|
|
|
self.nplots = 1
|
|
|
|
|
|
if showprofile:
|
|
|
self.nplots += 1
|
|
|
|
|
|
self.colorbar = colorbar
|
|
|
self.showprofile = showprofile
|
|
|
self.createPlots()
|
|
|
|
|
|
def createPlots(self):
|
|
|
plotObjList = []
|
|
|
idplot = 0
|
|
|
xi, yi, xw, yw = self.getScreenPos(idplot)
|
|
|
plotObj = SpcPlot(self.drvObj, self.idframe, idplot, xi, yi, xw, yw, self.colorbar)
|
|
|
plotObjList.append(plotObj)
|
|
|
|
|
|
if self.showprofile:
|
|
|
idplot = 1
|
|
|
xi, yi, xw, yw = self.getScreenPos(idplot)
|
|
|
type = "pwbox"
|
|
|
title = ""
|
|
|
xlabel = "dB"
|
|
|
ylabel = ""
|
|
|
plotObj = Plot1D(self.drvObj, self.idframe, idplot, xi, yi, xw, yw, type, title, xlabel, ylabel)
|
|
|
plotObjList.append(plotObj)
|
|
|
|
|
|
self.plotObjList = plotObjList
|
|
|
|
|
|
def getScreenPosMainPlot(self):
|
|
|
xi = 0.15
|
|
|
|
|
|
if self.showprofile:
|
|
|
xw = 0.65
|
|
|
|
|
|
else:
|
|
|
xw = 0.75
|
|
|
|
|
|
if self.colorbar:
|
|
|
xw = xw - 0.06
|
|
|
|
|
|
yi = 0.20; yw = 0.75
|
|
|
|
|
|
return xi, yi, xw, yw
|
|
|
|
|
|
def getScreenPosGraph1(self):
|
|
|
if self.colorbar:
|
|
|
xi = 0.65 + 0.08
|
|
|
else:
|
|
|
xi = 0.75 + 0.05
|
|
|
|
|
|
xw = xi + 0.2
|
|
|
|
|
|
yi = 0.2; yw = 0.75
|
|
|
|
|
|
return xi, yi, xw, yw
|
|
|
|
|
|
def plot(self,x, y, data):
|
|
|
plotObj = self.plotObjList[0]
|
|
|
plotObj.plot(x,y,data)
|
|
|
|
|
|
if self.showprofile:
|
|
|
plotObj = self.plotObjList[1]
|
|
|
plotObj.plot(data,y)
|
|
|
|
|
|
class SpcPlot(Plot):
|
|
|
|
|
|
getGrid = True
|
|
|
|
|
|
def __init__(self, drvObj, idframe, idplot, xi, yi, xw, yw, colorbar):
|
|
|
self.drvObj = drvObj
|
|
|
self.idframe = idframe
|
|
|
self.idplot = idplot
|
|
|
self.xi = xi
|
|
|
self.yi = yi
|
|
|
self.xw = xw
|
|
|
self.yw = yw
|
|
|
self.colorbar = colorbar
|
|
|
|
|
|
if self.colorbar:
|
|
|
cbxi = xw + 0.03
|
|
|
cbxw = cbxi + 0.03
|
|
|
cbyi = yi
|
|
|
cbyw = yw
|
|
|
self.cbxpos = [cbxi,cbxw]
|
|
|
self.cbypos = [cbyi,cbyw]
|
|
|
|
|
|
self.xpos = [self.xi,self.xw]
|
|
|
self.ypos = [self.yi,self.yw]
|
|
|
self.xaxisastime = False
|
|
|
self.timefmt = None
|
|
|
self.xopt = "bcnst"
|
|
|
self.yopt = "bcnstv"
|
|
|
|
|
|
self.szchar = 0.8
|
|
|
self.strforchannel = "Channel %d"%self.idframe
|
|
|
self.xlabel = "m/s"
|
|
|
self.ylabel = "Range (Km)"
|
|
|
|
|
|
def setBox(self, xmin, xmax, ymin, ymax, minvalue, maxvalue, *args):
|
|
|
self.xmin = xmin
|
|
|
self.xmax = xmax
|
|
|
self.ymin = ymin
|
|
|
self.ymax = ymax
|
|
|
self.minvalue = minvalue
|
|
|
self.maxvalue = maxvalue
|
|
|
self.colorbar = args[2]
|
|
|
self.title = "%s - %s"%(self.strforchannel,args[3])
|
|
|
|
|
|
|
|
|
|
|
|
def plot(self, x, y, data):
|
|
|
z = data
|
|
|
deltax = None
|
|
|
deltay = None
|
|
|
self.plotPcolor(x, y, z, deltax, deltay, self.getGrid)
|
|
|
self.getGrid = False
|
|
|
|
|
|
|
|
|
class RTIFigure(Figure):
|
|
|
overplot = 1
|
|
|
xw = 700
|
|
|
yw = 650
|
|
|
showprofile = False
|
|
|
starttime = None
|
|
|
endtime = None
|
|
|
minrange = None
|
|
|
maxrange = None
|
|
|
minvalue = None
|
|
|
maxvalue = None
|
|
|
xrangestepinsecs = None
|
|
|
|
|
|
|
|
|
def __init__(self, idfigure, nframes, wintitle, driver, colormap="br_green", colorbar= True, showprofile=False):
|
|
|
Figure.__init__(self,idfigure, nframes, wintitle, self.xw, self.yw, self.overplot, driver, colormap, colorbar)
|
|
|
|
|
|
self.showprofile = showprofile
|
|
|
|
|
|
def getSubplots(self):
|
|
|
nrows = self.nframes
|
|
|
ncolumns = 1
|
|
|
return nrows, ncolumns
|
|
|
|
|
|
def setParms(self, data, x, y, xmin, xmax, ymin, ymax, minvalue, maxvalue, xrangestep, deltax):
|
|
|
|
|
|
self.starttime = xmin
|
|
|
self.endtime = xmax
|
|
|
|
|
|
cdatetime = datetime.datetime.utcfromtimestamp(x)
|
|
|
|
|
|
mindatetime = datetime.datetime(cdatetime.year,cdatetime.month,cdatetime.day,self.starttime.hour,self.starttime.minute,self.starttime.second)
|
|
|
if ((xrangestep == 0) or (xrangestep == None)):
|
|
|
maxdatetime = datetime.datetime(cdatetime.year,cdatetime.month,cdatetime.day,self.endtime.hour,self.endtime.minute,self.endtime.second)
|
|
|
self.xrangestepinsecs = time.mktime(maxdatetime.timetuple()) - time.mktime(mindatetime.timetuple())
|
|
|
npoints = 1000.
|
|
|
if xrangestep == 1:
|
|
|
maxdatetime = mindatetime + datetime.timedelta(hours=1)
|
|
|
self.xrangestepinsecs = 60*60.
|
|
|
npoints = 500.
|
|
|
if xrangestep == 2:
|
|
|
maxdatetime = mindatetime + datetime.timedelta(minutes=1)
|
|
|
self.xrangestepinsecs = 60.
|
|
|
npoints = 250.
|
|
|
if xrangestep == 3:
|
|
|
maxdatetime = mindatetime + datetime.timedelta(seconds=1)
|
|
|
self.xrangestepinsecs = 1.
|
|
|
npoints = 125.
|
|
|
|
|
|
xmin = time.mktime(mindatetime.timetuple())
|
|
|
xmax = time.mktime(maxdatetime.timetuple())
|
|
|
|
|
|
deltax1 = (xmax-xmin) / npoints
|
|
|
# deltax = timeInterval
|
|
|
|
|
|
|
|
|
if ymin == None: ymin = numpy.min(y)
|
|
|
if ymax == None: ymax = numpy.max(y)
|
|
|
|
|
|
if minvalue == None: minvalue = 0.
|
|
|
if maxvalue == None: maxvalue = 50.
|
|
|
|
|
|
self.xmin = xmin
|
|
|
self.xmax = xmax
|
|
|
self.minrange = ymin
|
|
|
self.maxrange = ymax
|
|
|
self.ymin = ymin
|
|
|
self.ymax = ymax
|
|
|
self.minvalue = minvalue
|
|
|
self.maxvalue = maxvalue
|
|
|
self.xrangestep = xrangestep
|
|
|
self.deltax = deltax
|
|
|
|
|
|
def changeXRange(self,x):
|
|
|
|
|
|
cdatetime = datetime.datetime.utcfromtimestamp(x)
|
|
|
|
|
|
if ((cdatetime.time()>=self.starttime) and (cdatetime.time()<self.endtime)):
|
|
|
if self.xrangestep == 1:
|
|
|
mindatetime = datetime.datetime(cdatetime.year,cdatetime.month,cdatetime.day,cdatetime.hour,self.starttime.minute,self.starttime.second)
|
|
|
|
|
|
if self.xrangestep == 2:
|
|
|
mindatetime = datetime.datetime(cdatetime.year,cdatetime.month,cdatetime.day,cdatetime.hour,cdatetime.minute,self.starttime.second)
|
|
|
|
|
|
if self.xrangestep == 3:
|
|
|
mindatetime = datetime.datetime(cdatetime.year,cdatetime.month,cdatetime.day,cdatetime.hour,cdatetime.minute,cdatetime.second)
|
|
|
|
|
|
self.xmin = time.mktime(mindatetime.timetuple()) - time.timezone
|
|
|
self.xmax = self.xmin + self.xrangestepinsecs
|
|
|
|
|
|
self.figuretitle = "%s %s : %s"%(self.figuretitle,
|
|
|
datetime.datetime.utcfromtimestamp(self.xmin).strftime("%d-%b-%Y %H:%M:%S"),
|
|
|
datetime.datetime.utcfromtimestamp(self.xmax).strftime("%d-%b-%Y %H:%M:%S"))
|
|
|
return 1
|
|
|
|
|
|
return 0
|
|
|
|
|
|
def createFrames(self):
|
|
|
for frame in range(self.nframes):
|
|
|
frameObj = RTIFrame(self.drvObj,frame + 1, self.colorbar, self.showprofile)
|
|
|
self.frameObjList.append(frameObj)
|
|
|
|
|
|
class RTIFrame(Frame):
|
|
|
def __init__(self,drvObj,idframe,colorbar,showprofile):
|
|
|
self.drvObj = drvObj
|
|
|
self.idframe = idframe
|
|
|
self.nplots = 1
|
|
|
|
|
|
if showprofile:
|
|
|
self.nplots += 1
|
|
|
|
|
|
self.colorbar = colorbar
|
|
|
self.showprofile = showprofile
|
|
|
self.createPlots()
|
|
|
|
|
|
def createPlots(self):
|
|
|
plotObjList = []
|
|
|
|
|
|
idplot = 0
|
|
|
xi, yi, xw, yw = self.getScreenPos(idplot)
|
|
|
|
|
|
plotObj = RTIPlot(self.drvObj, self.idframe, idplot, xi, yi, xw, yw, self.colorbar)
|
|
|
plotObjList.append(plotObj)
|
|
|
|
|
|
if self.showprofile:
|
|
|
idplot = 1
|
|
|
xi, yi, xw, yw = self.getScreenPos(idplot)
|
|
|
type = "pwbox"
|
|
|
title = ""
|
|
|
xlabel = "dB"
|
|
|
ylabel = ""
|
|
|
plotObj = Plot1D(self.drvObj, self.idframe, idplot, xi, yi, xw, yw, type, title, xlabel, ylabel)
|
|
|
plotObjList.append(plotObj)
|
|
|
|
|
|
self.plotObjList = plotObjList
|
|
|
|
|
|
def getScreenPosMainPlot(self):
|
|
|
xi = 0.07
|
|
|
if self.showprofile:
|
|
|
xw = 0.65
|
|
|
else:
|
|
|
xw = 0.9
|
|
|
|
|
|
if self.colorbar:
|
|
|
xw = xw - 0.06
|
|
|
|
|
|
yi = 0.20; yw = 0.75
|
|
|
|
|
|
return xi, yi, xw, yw
|
|
|
|
|
|
def getScreenPosGraph1(self):
|
|
|
if self.colorbar:
|
|
|
xi = 0.65 + 0.08
|
|
|
else:
|
|
|
xi = 0.9 + 0.05
|
|
|
|
|
|
xw = xi + 0.2
|
|
|
|
|
|
yi = 0.2; yw = 0.75
|
|
|
|
|
|
return xi, yi, xw, yw
|
|
|
|
|
|
def plot(self, currenttime, range, data):
|
|
|
plotObj = self.plotObjList[0]
|
|
|
plotObj.plot(currenttime,range,data)
|
|
|
|
|
|
if self.showprofile:
|
|
|
plotObj = self.plotObjList[1]
|
|
|
plotObj.plot(data,range)
|
|
|
|
|
|
|
|
|
class RTIPlot(Plot):
|
|
|
deltax = None
|
|
|
deltay = None
|
|
|
xrange = [None,None]
|
|
|
xminpos = None
|
|
|
xmaxpos = None
|
|
|
xg = None
|
|
|
yg = None
|
|
|
|
|
|
def __init__(self,drvObj, idframe, idplot, xi, yi, xw, yw, colorbar):
|
|
|
self.drvObj = drvObj
|
|
|
self.idframe = idframe
|
|
|
self.idplot = idplot
|
|
|
self.xi = xi
|
|
|
self.yi = yi
|
|
|
self.xw = xw
|
|
|
self.yw = yw
|
|
|
self.colorbar = colorbar
|
|
|
|
|
|
if self.colorbar:
|
|
|
cbxi = xw + 0.03
|
|
|
cbxw = cbxi + 0.03
|
|
|
cbyi = yi
|
|
|
cbyw = yw
|
|
|
self.cbxpos = [cbxi,cbxw]
|
|
|
self.cbypos = [cbyi,cbyw]
|
|
|
|
|
|
self.xpos = [self.xi,self.xw]
|
|
|
self.ypos = [self.yi,self.yw]
|
|
|
self.xaxisastime = True
|
|
|
self.timefmt = "%H:%M"
|
|
|
self.xopt = "bcnstd"
|
|
|
self.yopt = "bcnstv"
|
|
|
|
|
|
self.szchar = 1.0
|
|
|
self.title = "Channel %d"%self.idframe
|
|
|
self.xlabel = "Local Time"
|
|
|
self.ylabel = "Range (Km)"
|
|
|
|
|
|
|
|
|
def setBox(self, xmin, xmax, ymin, ymax, minvalue, maxvalue, deltax=None, deltay=None, colorbar=True, *args):
|
|
|
self.xmin = xmin
|
|
|
self.xmax = xmax
|
|
|
self.ymin = ymin
|
|
|
self.ymax = ymax
|
|
|
self.minvalue = minvalue
|
|
|
self.maxvalue = maxvalue
|
|
|
self.deltax = deltax
|
|
|
self.deltay = deltay
|
|
|
self.colorbar = colorbar
|
|
|
|
|
|
def plot(self, currenttime, range, data):
|
|
|
|
|
|
if self.xmaxpos == None:
|
|
|
self.xmaxpos = currenttime
|
|
|
|
|
|
# if currenttime >= self.xmaxpos:
|
|
|
|
|
|
self.xminpos = currenttime
|
|
|
self.xmaxpos = currenttime + self.deltax
|
|
|
x = [currenttime]
|
|
|
y = range
|
|
|
z = numpy.reshape(data, (1,-1))
|
|
|
getGrid = True
|
|
|
|
|
|
self.plotPcolor(x, y, z, self.deltax, self.deltay, getGrid)
|
|
|
|
|
|
|
|
|
class ScopeFigure(Figure):
|
|
|
overplot = 0
|
|
|
xw = 700
|
|
|
yw = 650
|
|
|
colorbar = None
|
|
|
|
|
|
def __init__(self,idfigure,nframes,wintitle,driver):
|
|
|
colormap = None
|
|
|
colorbar = False
|
|
|
|
|
|
Figure.__init__(self,idfigure, nframes, wintitle, self.xw, self.yw, self.overplot, driver, colormap, colorbar)
|
|
|
|
|
|
|
|
|
def getSubplots(self):
|
|
|
nrows = self.nframes
|
|
|
ncolumns = 1
|
|
|
return nrows, ncolumns
|
|
|
|
|
|
def createFrames(self):
|
|
|
for frame in range(self.nframes):
|
|
|
frameObj = ScopeFrame(self.drvObj,frame + 1)
|
|
|
self.frameObjList.append(frameObj)
|
|
|
|
|
|
|
|
|
class ScopeFrame(Frame):
|
|
|
# plotObjList = []
|
|
|
xlabel = ""
|
|
|
ylabel = ""
|
|
|
title = ""
|
|
|
def __init__(self,drvObj,idframe):
|
|
|
self.drvObj = drvObj
|
|
|
self.idframe = idframe
|
|
|
self.nplots = 1 #nplots/frame
|
|
|
self.createPlots()
|
|
|
# Frame.__init__(self, drvObj, idframe)
|
|
|
|
|
|
def getScreenPosMainPlot(self):#cada Frame determina las coordenadas de los plots
|
|
|
xi = 0.08; xw = 0.9
|
|
|
yi = 0.20; yw = 0.75
|
|
|
return xi,yi,xw,yw
|
|
|
|
|
|
def createPlots(self):
|
|
|
plotObjList = []
|
|
|
for idplot in range(self.nplots):
|
|
|
xi, yi, xw, yw = self.getScreenPos(idplot)
|
|
|
type = "scopebox"
|
|
|
title = "Channel %d"%self.idframe
|
|
|
xlabel = "range (Km)"
|
|
|
ylabel = "intensity"
|
|
|
plotObj = Plot1D(self.drvObj, self.idframe, idplot, xi, yi, xw, yw, type, title, xlabel, ylabel)
|
|
|
plotObjList.append(plotObj)
|
|
|
self.plotObjList = plotObjList
|
|
|
# self.plotObjList.append(plotObj)
|
|
|
|
|
|
def plot(self, x, y, z=None):
|
|
|
for plotObj in self.plotObjList:
|
|
|
plotObj.plot(x, y)
|
|
|
|
|
|
|
|
|
class Plot1D(Plot):
|
|
|
# type, title, xlabel, ylabel
|
|
|
def __init__(self, drvObj, idframe, idplot, xi, yi, xw, yw, type, title, xlabel, ylabel):
|
|
|
self.drvObj = drvObj
|
|
|
self.idframe = idframe
|
|
|
self.idplot = idplot
|
|
|
self.xi = xi
|
|
|
self.yi = yi
|
|
|
self.xw = xw
|
|
|
self.yw = yw
|
|
|
self.xpos = [self.xi,self.xw]
|
|
|
self.ypos = [self.yi,self.yw]
|
|
|
self.xaxisastime = False
|
|
|
self.timefmt = None
|
|
|
self.xopt = "bcnst"
|
|
|
self.yopt = "bcnstv"
|
|
|
self.szchar = 1.0
|
|
|
self.type = type
|
|
|
self.title = title
|
|
|
self.xlabel = xlabel
|
|
|
self.ylabel = ylabel
|
|
|
|
|
|
|
|
|
|
|
|
def setBox(self, xmin, xmax, ymin, ymax, minvalue, maxvalue, *args):
|
|
|
if self.type == "pwbox":
|
|
|
self.xmin = minvalue
|
|
|
self.xmax = maxvalue
|
|
|
self.ymin = ymin
|
|
|
self.ymax = ymax
|
|
|
self.minvalue = minvalue
|
|
|
self.maxvalue = maxvalue
|
|
|
|
|
|
else:
|
|
|
self.xmin = xmin
|
|
|
self.xmax = xmax
|
|
|
self.ymin = ymin
|
|
|
self.ymax = ymax
|
|
|
self.minvalue = minvalue
|
|
|
self.maxvalue = maxvalue
|
|
|
|
|
|
self.colorbar = False
|
|
|
|
|
|
def plot(self,x,y):
|
|
|
if y.dtype == "complex128":
|
|
|
color="blue"
|
|
|
self.plotBasicLine(x, y.real, color)
|
|
|
color="red"
|
|
|
self.plotBasicLine(x, y.imag, color)
|
|
|
else:
|
|
|
color="blue"
|
|
|
self.plotBasicLine(x, y, color)
|
|
|
|
