|
|
import mpldriver
|
|
|
|
|
|
class Figure:
|
|
|
axesList = None
|
|
|
width = None
|
|
|
height = None
|
|
|
def __init__(self):
|
|
|
pass
|
|
|
|
|
|
def init(self, idfigure, wintitle, width, height, nplots):
|
|
|
self.idfigure = idfigure
|
|
|
self.wintitle = wintitle
|
|
|
self.width = width
|
|
|
self.height = height
|
|
|
self.nplots = nplots
|
|
|
self.fig = mpldriver.init(idfigure, wintitle, width, height)
|
|
|
|
|
|
self.axesList = []
|
|
|
|
|
|
def setTitle(self, title):
|
|
|
mpldriver.setTitle(self.idfigure, title)
|
|
|
|
|
|
def setWinTitle(self,title):
|
|
|
mpldriver.setWinTitle(fig=self.fig, title=title)
|
|
|
|
|
|
def setTextFromAxes(self, title):
|
|
|
mpldriver.setTextFromAxes(self.idfigure, self.axesList[0].ax, title)
|
|
|
|
|
|
def makeAxes(self, nrow, ncol, xpos, ypos, colspan, rowspan):
|
|
|
ax = mpldriver.makeAxes(self.idfigure, nrow, ncol, xpos, ypos, colspan, rowspan)
|
|
|
axesObj = Axes(ax)
|
|
|
self.axesList.append(axesObj)
|
|
|
|
|
|
def draw(self):
|
|
|
mpldriver.draw(self.idfigure)
|
|
|
|
|
|
def run(self):
|
|
|
pass
|
|
|
|
|
|
|
|
|
class Axes:
|
|
|
firsttime = None
|
|
|
ax = None
|
|
|
mesh = None
|
|
|
|
|
|
def __init__(self, ax):
|
|
|
self.firsttime = True
|
|
|
self.ax = ax
|
|
|
self.mesh = None
|
|
|
|
|
|
def pline(self, x, y, xmin, xmax, ymin, ymax, xlabel, ylabel, title):
|
|
|
|
|
|
mpldriver.pline(ax=self.ax,
|
|
|
x=x,
|
|
|
y=y,
|
|
|
xmin=xmin,
|
|
|
xmax=xmax,
|
|
|
ymin=ymin,
|
|
|
ymax=ymax,
|
|
|
xlabel=xlabel,
|
|
|
ylabel=ylabel,
|
|
|
title=title,
|
|
|
firsttime=self.firsttime)
|
|
|
|
|
|
self.firsttime = False
|
|
|
|
|
|
def pcolor(self, x, y, z, xmin, xmax, ymin, ymax, zmin, zmax, xlabel, ylabel, title):
|
|
|
meshfromaxes=mpldriver.pcolor(ax=self.ax,
|
|
|
x=x,
|
|
|
y=y,
|
|
|
z=z,
|
|
|
xmin=xmin,
|
|
|
xmax=xmax,
|
|
|
ymin=ymin,
|
|
|
ymax=ymax,
|
|
|
zmin=zmin,
|
|
|
zmax=zmax,
|
|
|
xlabel=xlabel,
|
|
|
ylabel=ylabel,
|
|
|
title=title,
|
|
|
firsttime=self.firsttime,
|
|
|
mesh=self.mesh)
|
|
|
self.mesh = meshfromaxes
|
|
|
self.firsttime = False
|
|
|
|
