jroplot.py
181 lines
| 5.4 KiB
| text/x-python
|
PythonLexer
Daniel Valdez
|
r190 | import numpy | |
| import datetime | |||
| from graphics.figure import * | |||
|
Daniel Valdez
|
r192 | class SpectraPlot(Figure): | |
| __isConfig = None | |||
| def __init__(self): | |||
| self.__isConfig = False | |||
| self.width = 850 | |||
| self.height = 800 | |||
| def getSubplots(self): | |||
| ncol = int(numpy.sqrt(self.nplots)+0.9) | |||
| nrow = int(self.nplots*1./ncol + 0.9) | |||
| return nrow, ncol | |||
| def setAxesWithOutProfiles(self, nrow, ncol): | |||
| colspan = 1 | |||
| rowspan = 1 | |||
| counter = 0 | |||
| for y in range(nrow): | |||
| for x in range(ncol): | |||
| if counter < self.nplots: | |||
| # plt.subplot2grid((nrow, ncol), (y, x), colspan=colspan, rowspan=rowspan) | |||
| self.makeAxes(nrow, ncol, y, x, colspan, rowspan) | |||
| counter += 1 | |||
| def setAxesWithProfiles(self, nrow, ncol): | |||
| colspan = 1 | |||
| rowspan = 1 | |||
| factor = 2 | |||
| ncol = ncol*factor | |||
| counter = 0 | |||
| for y in range(nrow): | |||
| for x in range(ncol): | |||
| if counter < self.nplots*factor: | |||
| # plt.subplot2grid((nrow, ncol), (y, x), colspan=colspan, rowspan=rowspan) | |||
| self.makeAxes(nrow, ncol, y, x, colspan, rowspan) | |||
| counter += 1 | |||
| def setup(self, idfigure, wintitle, width, height, nplots, profile): | |||
| self.init(idfigure, wintitle, width, height, nplots) | |||
| nrow,ncol = self.getSubplots() | |||
| if profile: | |||
| self.setAxesWithProfiles(nrow, ncol) | |||
| else: | |||
| self.setAxesWithOutProfiles(nrow, ncol) | |||
| def run(self, dataOut, idfigure, wintitle="", channelList=None, xmin=None, xmax=None, ymin=None, ymax=None, zmin=None, zmax=None, profile=False): | |||
| if dataOut.isEmpty(): | |||
| return None | |||
| if channelList == None: | |||
| channelList = dataOut.channelList | |||
| nplots = len(channelList) | |||
| z = 10.*numpy.log10(dataOut.data_spc[channelList,:,:]) | |||
| y = dataOut.heightList | |||
| x = numpy.arange(dataOut.nFFTPoints) | |||
| if not self.__isConfig: | |||
| self.setup(idfigure=idfigure, | |||
| wintitle=wintitle, | |||
| width=self.width, | |||
| height=self.height, | |||
| nplots=nplots, | |||
| profile=profile) | |||
| if xmin == None: self.xmin = numpy.min(x) | |||
| if xmax == None: self.xmax = numpy.max(x) | |||
| if ymin == None: self.ymin = numpy.min(y) | |||
| if ymax == None: self.ymax = numpy.max(y) | |||
| if zmin == None: self.zmin = 0 | |||
| if zmax == None: self.zmax = 90 | |||
| self.__isConfig = True | |||
| ylabel = "Range[Km]" | |||
| xlabel = "m/s" | |||
| for i in range(len(self.axesList)): | |||
| title = "Channel %d"%i | |||
| axes = self.axesList[i] | |||
| z2 = z[i,:,:] | |||
| axes.pcolor(x, y, z, self.xmin, self.xmax, self.ymin, self.ymax, self.zmin, self.zmax, xlabel, ylabel, title) | |||
| self.draw() | |||
|
Daniel Valdez
|
r190 | class Scope(Figure): | |
| __isConfig = None | |||
| def __init__(self): | |||
| self.__isConfig = False | |||
| self.width = 850 | |||
| self.height = 800 | |||
| def getSubplots(self): | |||
| nrow = self.nplots | |||
| ncol = 3 | |||
| return nrow, ncol | |||
| def setup(self, idfigure, wintitle, width, height, nplots): | |||
| self.init(idfigure, wintitle, width, height, nplots) | |||
| nrow,ncol = self.getSubplots() | |||
| colspan = 3 | |||
| rowspan = 1 | |||
| for i in range(nplots): | |||
| self.makeAxes(nrow, ncol, i, 0, colspan, rowspan) | |||
| def run(self, dataOut, idfigure, wintitle="", channelList=None, xmin=None, xmax=None, ymin=None, ymax=None): | |||
| if dataOut.isEmpty(): | |||
| return None | |||
| if channelList == None: | |||
| channelList = dataOut.channelList | |||
| nplots = len(channelList) | |||
| y = dataOut.data[channelList,:] * numpy.conjugate(dataOut.data[channelList,:]) | |||
| y = y.real | |||
| x = dataOut.heightList | |||
| if not self.__isConfig: | |||
| self.setup(idfigure=idfigure, | |||
|
Daniel Valdez
|
r192 | wintitle=wintitle, | |
|
Daniel Valdez
|
r190 | width=self.width, | |
| height=self.height, | |||
| nplots=nplots) | |||
| if xmin == None: self.xmin = numpy.min(x) | |||
| if xmax == None: self.xmax = numpy.max(x) | |||
| if ymin == None: self.ymin = numpy.min(y) | |||
| if ymax == None: self.ymax = numpy.max(y) | |||
| self.__isConfig = True | |||
| thisDatetime = datetime.datetime.fromtimestamp(dataOut.utctime) | |||
| dateTime = "%s"%(thisDatetime.strftime("%d-%b-%Y %H:%M:%S")) | |||
| date = "%s"%(thisDatetime.strftime("%d-%b-%Y")) | |||
| figuretitle = "Scope: " + dateTime | |||
| self.setTitle(title=figuretitle) | |||
| # self.setTextFromAxes(title=figuretitle) | |||
| ylabel = "Intensity" | |||
| xlabel = "Range[Km]" | |||
| for i in range(len(self.axesList)): | |||
| title = "Channel %d"%i | |||
| axes = self.axesList[i] | |||
| y2 = y[i,:] | |||
| axes.pline(x, y2, self.xmin, self.xmax, self.ymin, self.ymax, xlabel, ylabel, title) | |||
| self.draw() | |||