jroplot_heispectra.py
329 lines
| 9.9 KiB
| text/x-python
|
PythonLexer
|
r487 | ''' | ||
|
r568 | Created on Jul 9, 2014 | ||
|
r487 | |||
|
r568 | @author: roj-idl71 | ||
|
r487 | ''' | ||
import os | ||||
import datetime | ||||
import numpy | ||||
from figure import Figure, isRealtime | ||||
|
r789 | from plotting_codes import * | ||
|
r487 | |||
class SpectraHeisScope(Figure): | ||||
|
r897 | |||
|
r487 | isConfig = None | ||
__nsubplots = None | ||||
|
r897 | |||
|
r487 | WIDTHPROF = None | ||
HEIGHTPROF = None | ||||
PREFIX = 'spc' | ||||
|
r897 | |||
def __init__(self, **kwargs): | ||||
Figure.__init__(self, **kwargs) | ||||
|
r487 | self.isConfig = False | ||
self.__nsubplots = 1 | ||||
|
r897 | |||
|
r487 | self.WIDTH = 230 | ||
self.HEIGHT = 250 | ||||
self.WIDTHPROF = 120 | ||||
self.HEIGHTPROF = 0 | ||||
self.counter_imagwr = 0 | ||||
|
r897 | |||
|
r789 | self.PLOT_CODE = SPEC_CODE | ||
|
r897 | |||
|
r487 | def getSubplots(self): | ||
|
r897 | |||
|
r487 | ncol = int(numpy.sqrt(self.nplots)+0.9) | ||
nrow = int(self.nplots*1./ncol + 0.9) | ||||
|
r897 | |||
|
r487 | return nrow, ncol | ||
|
r897 | |||
|
r487 | def setup(self, id, nplots, wintitle, show): | ||
|
r897 | |||
|
r487 | showprofile = False | ||
self.__showprofile = showprofile | ||||
self.nplots = nplots | ||||
|
r897 | |||
|
r487 | ncolspan = 1 | ||
colspan = 1 | ||||
if showprofile: | ||||
ncolspan = 3 | ||||
colspan = 2 | ||||
self.__nsubplots = 2 | ||||
|
r897 | |||
|
r487 | self.createFigure(id = id, | ||
wintitle = wintitle, | ||||
widthplot = self.WIDTH + self.WIDTHPROF, | ||||
heightplot = self.HEIGHT + self.HEIGHTPROF, | ||||
show = show) | ||||
|
r897 | |||
|
r487 | nrow, ncol = self.getSubplots() | ||
|
r897 | |||
|
r487 | counter = 0 | ||
for y in range(nrow): | ||||
for x in range(ncol): | ||||
|
r897 | |||
|
r487 | if counter >= self.nplots: | ||
break | ||||
|
r897 | |||
|
r487 | self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan, colspan, 1) | ||
|
r897 | |||
|
r487 | if showprofile: | ||
self.addAxes(nrow, ncol*ncolspan, y, x*ncolspan+colspan, 1, 1) | ||||
|
r897 | |||
|
r487 | counter += 1 | ||
|
r897 | |||
|
r487 | def run(self, dataOut, id, wintitle="", channelList=None, | ||
xmin=None, xmax=None, ymin=None, ymax=None, save=False, | ||||
|
r568 | figpath='./', figfile=None, ftp=False, wr_period=1, show=True, | ||
|
r789 | server=None, folder=None, username=None, password=None, | ||
ftp_wei=0, exp_code=0, sub_exp_code=0, plot_pos=0): | ||||
|
r897 | |||
|
r487 | """ | ||
|
r897 | |||
|
r487 | Input: | ||
dataOut : | ||||
id : | ||||
wintitle : | ||||
channelList : | ||||
xmin : None, | ||||
xmax : None, | ||||
ymin : None, | ||||
ymax : None, | ||||
""" | ||||
|
r897 | |||
|
r487 | if dataOut.realtime: | ||
if not(isRealtime(utcdatatime = dataOut.utctime)): | ||||
print 'Skipping this plot function' | ||||
return | ||||
|
r897 | |||
|
r487 | 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)) | ||||
|
r897 | |||
|
r487 | # x = dataOut.heightList | ||
|
r897 | c = 3E8 | ||
|
r487 | deltaHeight = dataOut.heightList[1] - dataOut.heightList[0] | ||
#deberia cambiar para el caso de 1Mhz y 100KHz | ||||
x = numpy.arange(-1*dataOut.nHeights/2.,dataOut.nHeights/2.)*(c/(2*deltaHeight*dataOut.nHeights*1000)) | ||||
#para 1Mhz descomentar la siguiente linea | ||||
#x= x/(10000.0) | ||||
# y = dataOut.data[channelIndexList,:] * numpy.conjugate(dataOut.data[channelIndexList,:]) | ||||
# y = y.real | ||||
|
r496 | factor = dataOut.normFactor | ||
data = dataOut.data_spc / factor | ||||
datadB = 10.*numpy.log10(data) | ||||
|
r487 | y = datadB | ||
|
r897 | |||
|
r487 | #thisDatetime = dataOut.datatime | ||
|
r568 | thisDatetime = datetime.datetime.utcfromtimestamp(dataOut.getTimeRange()[0]) | ||
|
r487 | title = wintitle + " Scope: %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S")) | ||
xlabel = "" | ||||
#para 1Mhz descomentar la siguiente linea | ||||
#xlabel = "Frequency x 10000" | ||||
ylabel = "Intensity (dB)" | ||||
|
r897 | |||
|
r487 | if not self.isConfig: | ||
nplots = len(channelIndexList) | ||||
|
r897 | |||
|
r487 | self.setup(id=id, | ||
nplots=nplots, | ||||
wintitle=wintitle, | ||||
show=show) | ||||
|
r897 | |||
|
r487 | 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) | ||||
|
r897 | |||
|
r789 | self.FTP_WEI = ftp_wei | ||
self.EXP_CODE = exp_code | ||||
self.SUB_EXP_CODE = sub_exp_code | ||||
self.PLOT_POS = plot_pos | ||||
|
r897 | |||
|
r487 | self.isConfig = True | ||
|
r897 | |||
|
r487 | self.setWinTitle(title) | ||
|
r897 | |||
|
r487 | for i in range(len(self.axesList)): | ||
ychannel = y[i,:] | ||||
str_datetime = '%s %s'%(thisDatetime.strftime("%Y/%m/%d"),thisDatetime.strftime("%H:%M:%S")) | ||||
|
r649 | title = "Channel %d: %4.2fdB: %s" %(dataOut.channelList[channelIndexList[i]], numpy.max(ychannel), str_datetime) | ||
|
r487 | axes = self.axesList[i] | ||
axes.pline(x, ychannel, | ||||
xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax, | ||||
xlabel=xlabel, ylabel=ylabel, title=title, grid='both') | ||||
|
r897 | |||
|
r487 | self.draw() | ||
|
r897 | |||
|
r573 | self.save(figpath=figpath, | ||
figfile=figfile, | ||||
save=save, | ||||
ftp=ftp, | ||||
wr_period=wr_period, | ||||
thisDatetime=thisDatetime) | ||||
|
r487 | |||
class RTIfromSpectraHeis(Figure): | ||||
|
r897 | |||
|
r487 | isConfig = None | ||
__nsubplots = None | ||||
PREFIX = 'rtinoise' | ||||
|
r897 | |||
def __init__(self, **kwargs): | ||||
Figure.__init__(self, **kwargs) | ||||
|
r487 | self.timerange = 24*60*60 | ||
self.isConfig = False | ||||
self.__nsubplots = 1 | ||||
|
r897 | |||
|
r487 | self.WIDTH = 820 | ||
self.HEIGHT = 200 | ||||
self.WIDTHPROF = 120 | ||||
self.HEIGHTPROF = 0 | ||||
self.counter_imagwr = 0 | ||||
self.xdata = None | ||||
self.ydata = None | ||||
|
r494 | self.figfile = None | ||
|
r897 | |||
|
r789 | self.PLOT_CODE = RTI_CODE | ||
|
r897 | |||
|
r487 | def getSubplots(self): | ||
|
r897 | |||
|
r487 | ncol = 1 | ||
nrow = 1 | ||||
|
r897 | |||
|
r487 | return nrow, ncol | ||
|
r897 | |||
|
r487 | def setup(self, id, nplots, wintitle, showprofile=True, show=True): | ||
|
r897 | |||
|
r487 | self.__showprofile = showprofile | ||
self.nplots = nplots | ||||
|
r897 | |||
|
r487 | ncolspan = 7 | ||
colspan = 6 | ||||
self.__nsubplots = 2 | ||||
|
r897 | |||
|
r487 | self.createFigure(id = id, | ||
wintitle = wintitle, | ||||
widthplot = self.WIDTH+self.WIDTHPROF, | ||||
heightplot = self.HEIGHT+self.HEIGHTPROF, | ||||
show = show) | ||||
|
r897 | |||
|
r487 | nrow, ncol = self.getSubplots() | ||
|
r897 | |||
|
r487 | self.addAxes(nrow, ncol*ncolspan, 0, 0, colspan, 1) | ||
|
r897 | |||
|
r487 | def run(self, dataOut, id, wintitle="", channelList=None, showprofile='True', | ||
xmin=None, xmax=None, ymin=None, ymax=None, | ||||
timerange=None, | ||||
|
r568 | save=False, figpath='./', figfile=None, ftp=False, wr_period=1, show=True, | ||
|
r789 | server=None, folder=None, username=None, password=None, | ||
ftp_wei=0, exp_code=0, sub_exp_code=0, plot_pos=0): | ||||
|
r897 | |||
|
r487 | 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)) | ||||
|
r897 | |||
|
r487 | if timerange != None: | ||
self.timerange = timerange | ||||
|
r897 | |||
|
r487 | x = dataOut.getTimeRange() | ||
y = dataOut.getHeiRange() | ||||
|
r897 | |||
|
r498 | factor = dataOut.normFactor | ||
data = dataOut.data_spc / factor | ||||
|
r487 | data = numpy.average(data,axis=1) | ||
datadB = 10*numpy.log10(data) | ||||
|
r897 | |||
|
r487 | # factor = dataOut.normFactor | ||
# noise = dataOut.getNoise()/factor | ||||
# noisedB = 10*numpy.log10(noise) | ||||
|
r897 | |||
|
r487 | #thisDatetime = dataOut.datatime | ||
|
r568 | thisDatetime = datetime.datetime.utcfromtimestamp(dataOut.getTimeRange()[0]) | ||
|
r487 | title = wintitle + " RTI: %s" %(thisDatetime.strftime("%d-%b-%Y")) | ||
xlabel = "Local Time" | ||||
ylabel = "Intensity (dB)" | ||||
|
r897 | |||
|
r487 | if not self.isConfig: | ||
|
r897 | |||
|
r487 | nplots = 1 | ||
|
r897 | |||
|
r487 | self.setup(id=id, | ||
nplots=nplots, | ||||
wintitle=wintitle, | ||||
showprofile=showprofile, | ||||
show=show) | ||||
|
r897 | |||
|
r792 | self.tmin, self.tmax = self.getTimeLim(x, xmin, xmax) | ||
|
r897 | |||
|
r487 | if ymin == None: ymin = numpy.nanmin(datadB) | ||
if ymax == None: ymax = numpy.nanmax(datadB) | ||||
|
r897 | |||
|
r487 | self.name = thisDatetime.strftime("%Y%m%d_%H%M%S") | ||
self.isConfig = True | ||||
|
r494 | self.figfile = figfile | ||
|
r487 | self.xdata = numpy.array([]) | ||
self.ydata = numpy.array([]) | ||||
|
r897 | |||
|
r789 | self.FTP_WEI = ftp_wei | ||
self.EXP_CODE = exp_code | ||||
self.SUB_EXP_CODE = sub_exp_code | ||||
self.PLOT_POS = plot_pos | ||||
|
r897 | |||
|
r487 | self.setWinTitle(title) | ||
|
r897 | |||
|
r487 | # title = "RTI %s" %(thisDatetime.strftime("%d-%b-%Y")) | ||
title = "RTI - %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S")) | ||||
|
r897 | |||
|
r487 | legendlabels = ["channel %d"%idchannel for idchannel in channelList] | ||
axes = self.axesList[0] | ||||
|
r897 | |||
|
r487 | self.xdata = numpy.hstack((self.xdata, x[0:1])) | ||
|
r897 | |||
|
r487 | if len(self.ydata)==0: | ||
self.ydata = datadB[channelIndexList].reshape(-1,1) | ||||
else: | ||||
self.ydata = numpy.hstack((self.ydata, datadB[channelIndexList].reshape(-1,1))) | ||||
|
r897 | |||
|
r487 | axes.pmultilineyaxis(x=self.xdata, y=self.ydata, | ||
|
r792 | xmin=self.tmin, xmax=self.tmax, ymin=ymin, ymax=ymax, | ||
|
r487 | xlabel=xlabel, ylabel=ylabel, title=title, legendlabels=legendlabels, marker='.', markersize=8, linestyle="solid", grid='both', | ||
XAxisAsTime=True | ||||
) | ||||
|
r897 | |||
|
r487 | self.draw() | ||
|
r897 | |||
|
r806 | update_figfile = False | ||
|
r897 | |||
|
r792 | if dataOut.ltctime >= self.tmax: | ||
|
r494 | self.counter_imagwr = wr_period | ||
|
r793 | self.isConfig = False | ||
|
r792 | update_figfile = True | ||
|
r897 | |||
|
r573 | self.save(figpath=figpath, | ||
figfile=figfile, | ||||
save=save, | ||||
ftp=ftp, | ||||
wr_period=wr_period, | ||||
thisDatetime=thisDatetime, | ||||
|
r792 | update_figfile=update_figfile) | ||