Add graphing support into pyfa (first iteration, very basic)

gui/graphFrame.py

@@ -20,21 +20,113 @@
 import wx
 try:
     import matplotlib as mpl
+    mpl.use('wxagg')
     from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as Canvas
+    from matplotlib.figure import Figure
+    import mpl_toolkits.axisartist as AA
+    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):
-        wx.Frame.__init__(self, parent)
+    def __init__(self, parent, style=wx.DEFAULT_FRAME_STYLE | wx.NO_FULL_REPAINT_ON_RESIZE):
+        wx.Frame.__init__(self, parent, style=style)
         self.mainSizer = wx.BoxSizer(wx.VERTICAL)
         self.SetSizer(self.mainSizer)
 
         self.graphSelection = wx.Choice(self, wx.ID_ANY, style=0)
         self.mainSizer.Add(self.graphSelection, 0, wx.EXPAND)
 
-        self.figure = mpl.figure.Figure(figsize=(4,2))
+        self.figure = Figure(figsize=(4, 3))
         self.canvas = Canvas(self, -1, self.figure)
+        self.subplot = self.figure.add_subplot(111)
 
         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.GetLabel()
+
+        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
+            return
+
+        x, y = success, status
+
+        self.subplot.plot(x, y)
+        self.canvas.draw()
+    def onFieldChanged(self, event):
+        try:
+            self.draw()
+        except:
+            pass