''' Created on Jul 9, 2014 @author: roj-idl71 ''' import os import datetime import numpy from schainpy.model.proc.jroproc_base import ProcessingUnit, Operation, MPDecorator #YONG from schainpy.utils import log from .figure import Figure @MPDecorator class Scope(Figure): isConfig = None def __init__(self):#, **kwargs): #YONG 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 dataOut.flagNoData: return dataOut 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) return dataOut