|
|
import numpy
|
|
|
from schainPlot import *
|
|
|
#from schainPlotLib import Driver
|
|
|
|
|
|
class ScopeFigure(Figure):
|
|
|
overplot = 0
|
|
|
xw = 700
|
|
|
yw = 650
|
|
|
colorbar = None
|
|
|
# frameObjList = []
|
|
|
|
|
|
def __init__(self,idfigure,nframes,wintitle,driver):
|
|
|
colormap = None
|
|
|
colorbar = False
|
|
|
addGraph = 0
|
|
|
args=(addGraph, addGraph)
|
|
|
|
|
|
|
|
|
self.idfigure = idfigure
|
|
|
self.nframes = nframes
|
|
|
self.wintitle = wintitle
|
|
|
# self.xw =
|
|
|
# self.yw =
|
|
|
# self.overplot =
|
|
|
self.driver = driver
|
|
|
self.colorbar = colorbar
|
|
|
self.colormap = colormap
|
|
|
|
|
|
self.drvObj = Driver(self.driver, self.idfigure, self.xw, self.yw, self.wintitle, self.overplot, self.colorbar, self.colormap)
|
|
|
self.drvObj.driver.setFigure()
|
|
|
|
|
|
# Figure.__init__(self,idfigure,nframes,wintitle,self.xw,self.yw,self.overplot,driver,colorbar,colormap,*args)
|
|
|
|
|
|
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.07; 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)
|
|
|
plotObj = Plot1D(self.drvObj, self.idframe, idplot, xi, yi, xw, yw)
|
|
|
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):
|
|
|
|
|
|
def __init__(self, drvObj, idframe, idplot, xi, yi, xw, yw):
|
|
|
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.title = "Channel %d"%self.idframe
|
|
|
self.xlabel = "x-axis"
|
|
|
self.ylabel = "y-axis"
|
|
|
|
|
|
|
|
|
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)
|
|
|
|
