from wxPython.wx import *
from Numeric import *
from NaCurves import *
class GraphPanel(wxScrolledWindow):
def __init__(self, parent, id=NewId(), pos=wxDefaultPosition, size=wxDefaultSize):
wxScrolledWindow.__init__(self, parent, id, pos, size)
self.normalFunc = NormalFunction()
self.lagrange = Lagrange(DATASET)
self.piecewiseLagrange = PiecewiseLagrange(DATASET, 4)
self.cubicSpline = Spline(DATASET)
self.errorLagrange = ErrorLagrange(DATASET)
self.errorPiecewiseLagrange = ErrorPiecewiseLagrange(DATASET, 4)
self.errorCubicSpline = ErrorSpline(DATASET)
EVT_PAINT(self, self.OnPaint)
def mappingToScreenX(self, x):
cx,cy = self.GetClientSizeTuple()
return (x) * (cx / 2) + cx / 2
def mappingToScreenY(self, y):
cx,cy = self.GetClientSizeTuple()
return (-y) * (cy / 2) + cy / 2
def OnPaint(self, event):
dc = wxPaintDC(self)
self.PrepareDC(dc)
dc.BeginDrawing()
self.drawBackground(dc)
self.drawGraph(dc)
#self.drawGuideLines(dc)
dc.EndDrawing()
def drawBackground(self, dc):
cx, cy = self.GetClientSizeTuple()
brush = wxBrush(wxColour(255,255,230), wxSOLID)
oldbrush = dc.GetBrush()
dc.SetBrush(brush)
dc.DrawRectangle(0,0, cx, cy)
dc.SetBrush(oldbrush)
def drawGuideLines(self, dc):
cx, cy = self.GetClientSizeTuple()
marginX = 100
marginY = 100
dc.DrawLine(marginX,marginY, marginX, cy-marginY)
dc.DrawLine(marginX,cy-marginY, cx-marginX, cy-marginY)
def drawGraph(self, dc):
self.plotNormalFunction(dc)
self.plotLagrange(dc)
self.plotPiecewiseLagrange(dc)
self.plotCubicSpline(dc)
self.plotErrorLagrange(dc)
self.plotErrorPiecewiseLagrange(dc)
self.plotErrorCubicSpline(dc)
def plotNormalFunction(self, dc):
self.plotGraph(dc, self.normalFunc, wxColour(255,0,0))
def plotGraph(self, dc, aFuncObject, aColor):
pen = wxPen(aColor, 1, wxSOLID)
oldpen = dc.GetPen()
dc.SetPen(pen)
for x in arange(-1.0, 1.0, 0.001):
y = aFuncObject.perform(x)
realX = self.mappingToScreenX(x)
realY = self.mappingToScreenY(y)
dc.DrawPoint(realX, realY)
dc.SetPen(oldpen)
def plotLagrange(self, dc):
self.plotGraph(dc, self.lagrange, wxColour(0,0,255))
def plotPiecewiseLagrange(self, dc):
self.plotGraph(dc, self.piecewiseLagrange, wxColour(255,255,0))
def plotCubicSpline(self, dc):
self.plotGraph(dc, self.cubicSpline, wxColour(0,0,0))
def plotErrorLagrange(self, dc):
self.plotGraph(dc, self.errorLagrange, wxColour(128,128,128))
def plotErrorPiecewiseLagrange(self, dc):
self.plotGraph(dc, self.errorPiecewiseLagrange, wxColour(255,255,200))
def plotErrorCubicSpline(self, dc):
self.plotGraph(dc, self.errorCubicSpline, wxColour(255,0,255))
