dave/pyfa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 12 Wiki Activity

Rework internal graph interfaces

Browse Source
This commit is contained in:
DarkPhoenix
2019-05-17 14:48:42 +03:00
parent 0b00e28863
commit f1384074b5
6 changed files with 127 additions and 203 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
gui/builtinGraphs/__init__.py
View File

@@ -1,13 +1,13 @@
# noinspection PyUnresolvedReferences
from gui.builtinGraphs import ( # noqa: E402,F401
fitDpsRange,
fitDpsTime,
fitDmgTime,
fitShieldRegenAmount,
fitShieldAmountTime,
fitCapRegenAmount,
# fitDpsRange,
# fitDpsTime,
# fitDmgTime,
# fitShieldRegenAmount,
# fitShieldAmountTime,
# fitCapRegenAmount,
fitCapAmountTime,
fitSpeedTime,
fitDistanceTime,
fitWarpTimeDistance
# fitSpeedTime,
# fitDistanceTime,
# fitWarpTimeDistance
)

62
gui/builtinGraphs/fitCapAmountTime.py
View File

@@ -17,65 +17,31 @@
# along with pyfa. If not, see <http://www.gnu.org/licenses/>.
# =============================================================================
import gui.mainFrame
from eos.graph import Data
from eos.graph.fitCapAmountTime import FitCapAmountTimeGraph as EosFitCapAmountTimeGraph
from gui.bitmap_loader import BitmapLoader
from gui.graph import Graph
from service.attribute import Attribute
from eos.graph.fitCapAmountTime import FitCapAmountTimeGraph as EosGraph
from gui.graph import Graph, XDef, YDef
class FitCapAmountTimeGraph(Graph):
propertyLabelMap = {"time": "Time (seconds)"}
defaults = EosFitCapAmountTimeGraph.defaults.copy()
name = 'Cap Amount vs Time'
def __init__(self):
Graph.__init__(self)
self.defaults["time"] = "0-300"
self.name = "Cap Amount vs Time"
self.eosGraph = None
self.eosGraph = EosGraph()
self.mainFrame = gui.mainFrame.MainFrame.getInstance()
def getFields(self):
return self.defaults
@property
def xDef(self):
return XDef(handle='time', inputDefault='0-300', inputLabel='Time (seconds)', inputIconID=1392, axisLabel='Time, s')
def getLabels(self):
return self.propertyLabelMap
@property
def yDefs(self):
return [YDef(handle='capAmount', switchLabel='Cap amount', axisLabel='Cap amount, GJ')]
def getIcons(self):
iconFile = Attribute.getInstance().getAttributeInfo('duration').iconID
bitmap = BitmapLoader.getBitmap(iconFile, "icons")
return {"time": bitmap}
def getPoints(self, fit, fields):
eosGraph = getattr(self, "eosGraph", None)
if eosGraph is None or eosGraph.fit != fit:
eosGraph = self.eosGraph = EosFitCapAmountTimeGraph(fit)
eosGraph.clearData()
variable = None
for fieldName, value in fields.items():
d = Data(fieldName, value)
if not d.isConstant():
if variable is None:
variable = fieldName
else:
# We can't handle more then one variable atm, OOPS FUCK OUT
return False, "Can only handle 1 variable"
eosGraph.setData(d)
if variable is None:
return False, "No variable"
x = []
y = []
for point, val in eosGraph.getIterator():
x.append(point[variable])
y.append(val)
return x, y
def getPlotPoints(self, fit, extraData, xRange, xAmount):
xRange = self.parseRange(xRange)
return self.eosGraph.getPlotPoints(fit, extraData, xRange, xAmount)
FitCapAmountTimeGraph.register()

47
gui/graph.py
View File

@@ -18,26 +18,59 @@
# =============================================================================
class Graph(object):
import re
from abc import ABCMeta, abstractmethod
from collections import namedtuple
class Graph(metaclass=ABCMeta):
views = []
yTypes = None
@classmethod
def register(cls):
Graph.views.append(cls)
def __init__(self):
self.name = ""
@property
@abstractmethod
def name(self):
raise NotImplementedError
def getFields(self, fit, fields):
raise NotImplementedError()
@property
@abstractmethod
def xDef(self):
raise NotImplementedError
def getIcons(self):
return None
@property
def extraInputs(self):
return ()
@property
@abstractmethod
def yDefs(self):
raise NotImplementedError
@property
def redrawOnEffectiveChange(self):
return False
def parseRange(self, string):
m = re.match('\s*(?P<first>\d+(\.\d+)?)\s*(-\s*(?P<second>\d+(\.\d+)?))?', string)
if m is None:
return (0, 0)
first = m.group('first')
second = m.group('second')
if not second:
return (0, float(first))
else:
return (float(first), float(second))
XDef = namedtuple('XDef', ('handle', 'inputDefault', 'inputLabel', 'inputIconID', 'axisLabel'))
YDef = namedtuple('YDef', ('handle', 'switchLabel', 'axisLabel'))
ExtraInput = namedtuple('ExtraInput', ('handle', 'inputDefault', 'inputLabel', 'inputIconID'))
# noinspection PyUnresolvedReferences
from gui.builtinGraphs import *

56
gui/graphFrame.py
View File

@@ -19,6 +19,7 @@
import os
import traceback
from itertools import chain
# noinspection PyPackageRequirements
import wx
@@ -230,50 +231,42 @@ class GraphFrame(wx.Frame):
def getValues(self):
values = {}
for fieldName, field in self.fields.items():
values[fieldName] = field.GetValue()
for fieldHandle, field in self.fields.items():
values[fieldHandle] = field.GetValue()
return values
def select(self, index):
view = self.getView()
icons = view.getIcons()
labels = view.getLabels()
sizer = self.gridSizer
sizer.Clear()
self.gridPanel.DestroyChildren()
self.fields.clear()
# Setup textboxes
for field, defaultVal in view.getFields().items():
for fieldDef in (view.xDef, *view.extraInputs):
textBox = wx.TextCtrl(self.gridPanel, wx.ID_ANY, style=0)
self.fields[field] = textBox
self.fields[fieldDef.handle] = textBox
textBox.Bind(wx.EVT_TEXT, self.onFieldChanged)
sizer.Add(textBox, 1, wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALL, 3)
if defaultVal is not None:
if not isinstance(defaultVal, str):
defaultVal = ("%f" % defaultVal).rstrip("0")
if defaultVal[-1:] == ".":
defaultVal += "0"
if fieldDef.inputDefault is not None:
inputDefault = fieldDef.inputDefault
if not isinstance(inputDefault, str):
inputDefault = ("%f" % inputDefault).rstrip("0")
if inputDefault[-1:] == ".":
inputDefault += "0"
textBox.ChangeValue(defaultVal)
textBox.ChangeValue(inputDefault)
imgLabelSizer = wx.BoxSizer(wx.HORIZONTAL)
if icons:
icon = icons.get(field)
if fieldDef.inputIconID:
icon = BitmapLoader.getBitmap(fieldDef.inputIconID, "icons")
if icon is not None:
static = wx.StaticBitmap(self.gridPanel)
static.SetBitmap(icon)
imgLabelSizer.Add(static, 0, wx.ALL | wx.ALIGN_CENTER_VERTICAL, 1)
if labels:
label = labels.get(field)
label = label if label is not None else field
else:
label = field
imgLabelSizer.Add(wx.StaticText(self.gridPanel, wx.ID_ANY, label), 0,
imgLabelSizer.Add(wx.StaticText(self.gridPanel, wx.ID_ANY, fieldDef.inputLabel), 0,
wx.LEFT | wx.RIGHT | wx.ALIGN_CENTER_VERTICAL, 3)
sizer.Add(imgLabelSizer, 0, wx.ALIGN_CENTER_VERTICAL)
sizer.Layout()
@@ -299,19 +292,20 @@ class GraphFrame(wx.Frame):
min_y = 0
max_y = 0
xRange = values[view.xDef.handle]
extraInputs = {i.handle: values[i.handle] for i in view.extraInputs}
chosenY = view.yDefs[0].handle
for fit in self.fits:
try:
success, status = view.getPoints(fit, values)
if not success:
# TODO: Add a pwetty statys bar to report errors with
self.SetStatusText(status)
return
xs, ys = view.getPlotPoints(fit, extraInputs, xRange, 100)
ys = ys[chosenY]
x, y = success, status
min_y = min(min_y, min(y, default=0))
max_y = max(max_y, max(y, default=0))
min_y = min(min_y, min(ys, default=0))
max_y = max(max_y, max(ys, default=0))
self.subplot.plot(x, y)
self.subplot.plot(xs, ys)
legend.append('{} ({})'.format(fit.name, fit.ship.item.getShortName()))
except Exception as ex:
pyfalog.warning("Invalid values in '{0}'", fit.name)