162 lines
5.4 KiB
Python
162 lines
5.4 KiB
Python
#===============================================================================
|
|
# Copyright (C) 2010 Diego Duclos
|
|
#
|
|
# This file is part of pyfa.
|
|
#
|
|
# pyfa is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# pyfa is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with pyfa. If not, see <http://www.gnu.org/licenses/>.
|
|
#===============================================================================
|
|
|
|
import wx
|
|
import gui.display
|
|
|
|
try:
|
|
import matplotlib as mpl
|
|
mpl.use('wxagg')
|
|
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as Canvas
|
|
from matplotlib.figure import Figure
|
|
enabled = True
|
|
except:
|
|
print "problems importing matplotlib, continueing without graphs"
|
|
Enabled = False
|
|
|
|
from gui.graph import Graph
|
|
from gui import bitmapLoader
|
|
|
|
class GraphFrame(wx.Frame):
|
|
def __init__(self, parent, style=wx.DEFAULT_FRAME_STYLE | wx.NO_FULL_REPAINT_ON_RESIZE):
|
|
wx.Frame.__init__(self, parent, style=style, size=(500, 500))
|
|
self.CreateStatusBar()
|
|
horSizer = wx.BoxSizer(wx.HORIZONTAL)
|
|
self.SetSizer(horSizer)
|
|
|
|
self.mainSizer = wx.BoxSizer(wx.VERTICAL)
|
|
horSizer.Add(self.mainSizer, 1, wx.EXPAND)
|
|
|
|
self.fitList = FitList(self)
|
|
self.fitList.SetMinSize((200, -1))
|
|
horSizer.Add(self.fitList, 0, wx.EXPAND)
|
|
|
|
self.graphSelection = wx.Choice(self, wx.ID_ANY, style=0)
|
|
self.mainSizer.Add(self.graphSelection, 0, wx.EXPAND)
|
|
|
|
self.figure = Figure(figsize=(4, 3))
|
|
self.canvas = Canvas(self, -1, self.figure)
|
|
self.subplot = self.figure.add_subplot(111)
|
|
self.subplot.grid(True)
|
|
|
|
self.mainSizer.Add(self.canvas, 0, wx.EXPAND)
|
|
|
|
self.gridPanel = wx.Panel(self)
|
|
self.mainSizer.Add(self.gridPanel, 1, wx.EXPAND)
|
|
|
|
dummyBox = wx.BoxSizer(wx.VERTICAL)
|
|
self.gridPanel.SetSizer(dummyBox)
|
|
|
|
self.gridSizer = wx.FlexGridSizer(0, 3)
|
|
self.gridSizer.AddGrowableCol(2)
|
|
dummyBox.Add(self.gridSizer, 0, wx.EXPAND)
|
|
|
|
for view in Graph.views:
|
|
view = view()
|
|
self.graphSelection.Append(view.name, view)
|
|
|
|
self.graphSelection.SetSelection(0)
|
|
self.fields = {}
|
|
self.select(0)
|
|
|
|
def getView(self):
|
|
return self.graphSelection.GetClientData(self.graphSelection.GetSelection())
|
|
|
|
def getValues(self):
|
|
values = {}
|
|
for fieldName, field in self.fields.iteritems():
|
|
values[fieldName] = field.GetValue()
|
|
|
|
return values
|
|
|
|
def select(self, index):
|
|
view = self.getView()
|
|
icons = view.getIcons()
|
|
labels = view.getLabels()
|
|
sizer = self.gridSizer
|
|
self.gridPanel.DestroyChildren()
|
|
self.fields.clear()
|
|
|
|
#Setup textboxes
|
|
for field, defaultVal in view.getFields().iteritems():
|
|
if icons:
|
|
icon = icons.get(field)
|
|
if icon is not None:
|
|
static = wx.StaticBitmap(self.gridPanel)
|
|
static.SetBitmap(icon)
|
|
sizer.Add(static, 0)
|
|
|
|
if labels:
|
|
label = labels.get(field)
|
|
label = label if label is not None else field
|
|
else:
|
|
label = field
|
|
|
|
sizer.Add(wx.StaticText(self.gridPanel, wx.ID_ANY, label), 0)
|
|
textBox = wx.TextCtrl(self.gridPanel, wx.ID_ANY, style=0)
|
|
self.fields[field] = textBox
|
|
textBox.Bind(wx.EVT_TEXT, self.onFieldChanged)
|
|
sizer.Add(textBox, 1, wx.EXPAND | wx.TOP, 2)
|
|
if defaultVal is not None:
|
|
if not isinstance(defaultVal, basestring):
|
|
defaultVal = ("%f" % defaultVal).rstrip("0")
|
|
if defaultVal[-1:] == ".":
|
|
defaultVal = defaultVal + "0"
|
|
|
|
textBox.ChangeValue(defaultVal)
|
|
|
|
self.draw()
|
|
|
|
def draw(self):
|
|
values = self.getValues()
|
|
view = self.getView()
|
|
success, status = view.getPoints(values)
|
|
if not success:
|
|
#TODO: Add a pwetty statys bar to report errors with
|
|
self.SetStatusText(status)
|
|
return
|
|
|
|
x, y = success, status
|
|
|
|
self.subplot.clear()
|
|
self.subplot.grid(True)
|
|
self.subplot.plot(x, y)
|
|
self.canvas.draw()
|
|
self.SetStatusText("")
|
|
|
|
def onFieldChanged(self, event):
|
|
try:
|
|
self.draw()
|
|
except:
|
|
self.SetStatusText("Invalid values")
|
|
|
|
class FitList(wx.Panel):
|
|
def __init__(self, parent):
|
|
wx.Panel.__init__(self, parent)
|
|
self.mainSizer = wx.BoxSizer(wx.VERTICAL)
|
|
self.SetSizer(self.mainSizer)
|
|
|
|
self.mainSizer.Add(wx.StaticText(self, wx.ID_ANY, "Fits: Drag fits onto the list to graph them"), 0, wx.ALIGN_CENTER_HORIZONTAL)
|
|
|
|
self.fitList = wx.ListCtrl(self)
|
|
self.mainSizer.Add(self.fitList, 1, wx.EXPAND)
|
|
|
|
class FitDisplay(gui.display.Display):
|
|
pass
|