jroplot_heispectra.py
324 lines
| 10.8 KiB
| text/x-python
|
PythonLexer
Daniel Valdez
|
r487 | ''' | ||
| @author: Daniel Suarez | ||||
| ''' | ||||
| import os | ||||
| import datetime | ||||
| import numpy | ||||
| from figure import Figure, isRealtime | ||||
| class SpectraHeisScope(Figure): | ||||
| isConfig = None | ||||
| __nsubplots = None | ||||
| WIDTHPROF = None | ||||
| HEIGHTPROF = None | ||||
| PREFIX = 'spc' | ||||
| def __init__(self): | ||||
| self.isConfig = False | ||||
| self.__nsubplots = 1 | ||||
| self.WIDTH = 230 | ||||
| self.HEIGHT = 250 | ||||
| self.WIDTHPROF = 120 | ||||
| self.HEIGHTPROF = 0 | ||||
| self.counter_imagwr = 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, id, nplots, wintitle, show): | ||||
| showprofile = False | ||||
| self.__showprofile = showprofile | ||||
| self.nplots = nplots | ||||
| ncolspan = 1 | ||||
| colspan = 1 | ||||
| if showprofile: | ||||
| ncolspan = 3 | ||||
| colspan = 2 | ||||
| self.__nsubplots = 2 | ||||
| self.createFigure(id = id, | ||||
| wintitle = wintitle, | ||||
| widthplot = self.WIDTH + self.WIDTHPROF, | ||||
| heightplot = self.HEIGHT + self.HEIGHTPROF, | ||||
| show = show) | ||||
| 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, id, wintitle="", channelList=None, | ||||
| xmin=None, xmax=None, ymin=None, ymax=None, save=False, | ||||
|
Daniel Valdez
|
r494 | figpath='', figfile=None, ftp=False, wr_period=1, show=True, | ||
|
Daniel Valdez
|
r487 | server=None, folder=None, username=None, password=None): | ||
| """ | ||||
| Input: | ||||
| dataOut : | ||||
| id : | ||||
| wintitle : | ||||
| channelList : | ||||
| xmin : None, | ||||
| xmax : None, | ||||
| ymin : None, | ||||
| ymax : None, | ||||
| """ | ||||
| if dataOut.realtime: | ||||
| if not(isRealtime(utcdatatime = dataOut.utctime)): | ||||
| print 'Skipping this plot function' | ||||
| return | ||||
| 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 | ||||
| c = 3E8 | ||||
| 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 | ||||
| datadB = 10.*numpy.log10(dataOut.data_spc) | ||||
| y = datadB | ||||
| #thisDatetime = dataOut.datatime | ||||
| thisDatetime = datetime.datetime.utcfromtimestamp(dataOut.getTimeRange()[1]) | ||||
| 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)" | ||||
| if not self.isConfig: | ||||
| nplots = len(channelIndexList) | ||||
| self.setup(id=id, | ||||
| nplots=nplots, | ||||
| wintitle=wintitle, | ||||
| show=show) | ||||
| 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)): | ||||
| ychannel = y[i,:] | ||||
| str_datetime = '%s %s'%(thisDatetime.strftime("%Y/%m/%d"),thisDatetime.strftime("%H:%M:%S")) | ||||
| title = "Channel %d: %4.2fdB: %s" %(i, numpy.max(ychannel), str_datetime) | ||||
| axes = self.axesList[i] | ||||
| axes.pline(x, ychannel, | ||||
| xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax, | ||||
| xlabel=xlabel, ylabel=ylabel, title=title, grid='both') | ||||
| self.draw() | ||||
|
Daniel Valdez
|
r494 | |||
| if figfile == None: | ||||
| str_datetime = thisDatetime.strftime("%Y%m%d_%H%M%S") | ||||
| figfile = self.getFilename(name = str_datetime) | ||||
| if figpath != '': | ||||
|
Daniel Valdez
|
r487 | self.counter_imagwr += 1 | ||
|
Daniel Valdez
|
r494 | if (self.counter_imagwr>=wr_period): | ||
| # store png plot to local folder | ||||
| self.saveFigure(figpath, figfile) | ||||
| # store png plot to FTP server according to RT-Web format | ||||
| #name = self.getNameToFtp(thisDatetime, self.FTP_WEI, self.EXP_CODE, self.SUB_EXP_CODE, self.PLOT_CODE, self.PLOT_POS) | ||||
| #ftp_filename = os.path.join(figpath, name) | ||||
| #self.saveFigure(figpath, ftp_filename) | ||||
|
Daniel Valdez
|
r487 | self.counter_imagwr = 0 | ||
| class RTIfromSpectraHeis(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.counter_imagwr = 0 | ||||
| self.xdata = None | ||||
| self.ydata = None | ||||
|
Daniel Valdez
|
r494 | self.figfile = None | ||
|
Daniel Valdez
|
r487 | |||
| def getSubplots(self): | ||||
| ncol = 1 | ||||
| nrow = 1 | ||||
| return nrow, ncol | ||||
| def setup(self, id, nplots, wintitle, showprofile=True, show=True): | ||||
| self.__showprofile = showprofile | ||||
| self.nplots = nplots | ||||
| ncolspan = 7 | ||||
| colspan = 6 | ||||
| self.__nsubplots = 2 | ||||
| self.createFigure(id = id, | ||||
| wintitle = wintitle, | ||||
| widthplot = self.WIDTH+self.WIDTHPROF, | ||||
| heightplot = self.HEIGHT+self.HEIGHTPROF, | ||||
| show = show) | ||||
| nrow, ncol = self.getSubplots() | ||||
| self.addAxes(nrow, ncol*ncolspan, 0, 0, colspan, 1) | ||||
| def run(self, dataOut, id, wintitle="", channelList=None, showprofile='True', | ||||
| xmin=None, xmax=None, ymin=None, ymax=None, | ||||
| timerange=None, | ||||
|
Daniel Valdez
|
r494 | save=False, figpath='', figfile=None, ftp=False, wr_period=1, show=True, | ||
|
Daniel Valdez
|
r487 | server=None, folder=None, username=None, password=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() | ||||
| data = dataOut.data_spc | ||||
| data = numpy.average(data,axis=1) | ||||
| datadB = 10*numpy.log10(data) | ||||
| # factor = dataOut.normFactor | ||||
| # noise = dataOut.getNoise()/factor | ||||
| # noisedB = 10*numpy.log10(noise) | ||||
| #thisDatetime = dataOut.datatime | ||||
| thisDatetime = datetime.datetime.utcfromtimestamp(dataOut.getTimeRange()[1]) | ||||
| title = wintitle + " RTI: %s" %(thisDatetime.strftime("%d-%b-%Y")) | ||||
| xlabel = "Local Time" | ||||
| ylabel = "Intensity (dB)" | ||||
| if not self.isConfig: | ||||
| nplots = 1 | ||||
| self.setup(id=id, | ||||
| nplots=nplots, | ||||
| wintitle=wintitle, | ||||
| showprofile=showprofile, | ||||
| show=show) | ||||
| tmin, tmax = self.getTimeLim(x, xmin, xmax) | ||||
| if ymin == None: ymin = numpy.nanmin(datadB) | ||||
| if ymax == None: ymax = numpy.nanmax(datadB) | ||||
| self.name = thisDatetime.strftime("%Y%m%d_%H%M%S") | ||||
| self.isConfig = True | ||||
|
Daniel Valdez
|
r494 | self.figfile = figfile | ||
|
Daniel Valdez
|
r487 | self.xdata = numpy.array([]) | ||
| self.ydata = numpy.array([]) | ||||
| self.setWinTitle(title) | ||||
| # title = "RTI %s" %(thisDatetime.strftime("%d-%b-%Y")) | ||||
| title = "RTI - %s" %(thisDatetime.strftime("%d-%b-%Y %H:%M:%S")) | ||||
| 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 = datadB[channelIndexList].reshape(-1,1) | ||||
| else: | ||||
| self.ydata = numpy.hstack((self.ydata, datadB[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='.', markersize=8, linestyle="solid", grid='both', | ||||
| XAxisAsTime=True | ||||
| ) | ||||
| self.draw() | ||||
|
Daniel Valdez
|
r494 | if x[1] >= self.axesList[0].xmax: | ||
| self.counter_imagwr = wr_period | ||||
|
Daniel Valdez
|
r487 | del self.xdata | ||
| del self.ydata | ||||
|
Daniel Valdez
|
r494 | self.__isConfig = False | ||
| if self.figfile == None: | ||||
| str_datetime = thisDatetime.strftime("%Y%m%d_%H%M%S") | ||||
| self.figfile = self.getFilename(name = str_datetime) | ||||
| if figpath != '': | ||||
| self.counter_imagwr += 1 | ||||
| if (self.counter_imagwr>=wr_period): | ||||
| # store png plot to local folder | ||||
| self.saveFigure(figpath, self.figfile) | ||||
| # store png plot to FTP server according to RT-Web format | ||||
| #name = self.getNameToFtp(thisDatetime, self.FTP_WEI, self.EXP_CODE, self.SUB_EXP_CODE, self.PLOT_CODE, self.PLOT_POS) | ||||
| #ftp_filename = os.path.join(figpath, name) | ||||
| #self.saveFigure(figpath, ftp_filename) | ||||
| self.counter_imagwr = 0 | ||||